序
大家好，我是王岳泽，天津大学研究生，数据竞赛爱好者。
我有幸取得了阿里云天池零基础入门数据挖掘-心跳信号分类预测第一名，特将经验分享给大家~ 但由于水平、精力有限，难免存在谬误，请大家海涵！
赛事地址，目前长期赛已经开放，大家都可以前往报名参与学习：
https://tianchi.aliyun.com/competition/entrance/531883/introduction
赛事经验原文地址，在天池实验室可以直接调试代码：
https://tianchi.aliyun.com/notebook-ai/detail?postId=231585
1. 依赖安装和导入
!pip install scikit-learn sklearn imblearn --user
import os import math import time import random import datetime import numpy as np import pandas as pd import seaborn as sns import matplotlib.pyplot as plt import tensorflow as tf import tensorflow.keras as K from tensorflow.keras import Sequential, utils, regularizers, Model, Input from tensorflow.keras.layers import Flatten, Dense, Conv1D, MaxPool1D, Dropout, AvgPool1D from imblearn.over_sampling import SMOTE from sklearn.model_selection import KFold from sklearn.preprocessing import OneHotEncoder
2. 数据加载与预处理
# 加载训练集和测试集(相对路径) train = pd.read_csv('./train.csv') test = pd.read_csv('./testA.csv')
# 数据精度量化压缩 def reduce_mem_usage(df): # 处理前 数据集总内存计算 start_mem = df.memory_usage().sum() / 1024**2 print('Memory usage of dataframe is {:.2f} MB'.format(start_mem)) # 遍历特征列 for col in df.columns: # 当前特征类型 col_type = df[col].dtype # 处理 numeric 型数据 if col_type != object: c_min = df[col].min() # 最小值 c_max = df[col].max() # 最大值 # int 型数据 精度转换 if str(col_type)[:3] == 'int': if c_min > np.iinfo(np.int8).min and c_max < np.iinfo(np.int8).max: df[col] = df[col].astype(np.int8) elif c_min > np.iinfo(np.int16).min and c_max < np.iinfo(np.int16).max: df[col] = df[col].astype(np.int16) elif c_min > np.iinfo(np.int32).min and c_max < np.iinfo(np.int32).max: df[col] = df[col].astype(np.int32) elif c_min > np.iinfo(np.int64).min and c_max < np.iinfo(np.int64).max: df[col] = df[col].astype(np.int64) # float 型数据 精度转换 else: if c_min > np.finfo(np.float16).min and c_max < np.finfo(np.float16).max: df[col] = df[col].astype(np.float16) elif c_min > np.finfo(np.float32).min and c_max < np.finfo(np.float32).max: df[col] = df[col].astype(np.float32) else: df[col] = df[col].astype(np.float64) # 处理 object 型数据 else: df[col] = df[col].astype('category') # object 转 category # 处理后 数据集总内存计算 end_mem = df.memory_usage().sum() / 1024**2 print('Memory usage after optimization is: {:.2f} MB'.format(end_mem)) print('Decreased by {:.1f}%'.format(100 * (start_mem - end_mem) / start_mem)) return df # 训练集特征处理与精度量化 train_list = [] for items in train.values: train_list.append([items[0]] + [float(i) for i in items[1].split(',')] + [items[2]]) train = pd.DataFrame(np.array(train_list)) train.columns = ['id'] + ['s_' + str(i) for i in range(len(train_list[0])-2)] + ['label'] # 特征分离 train = reduce_mem_usage(train) # 精度量化 # 测试集特征处理与精度量化 test_list=[] for items in test.values: test_list.append([items[0]] + [float(i) for i in items[1].split(',')]) test = pd.DataFrame(np.array(test_list)) test.columns = ['id'] + ['s_'+str(i) for i in range(len(test_list[0])-1)] # 特征分离 test = reduce_mem_usage(test) # 精度量化
# 查看训练集, 分离标签与样本, 去除 id y_train = train['label'] x_train = train.drop(['id', 'label'], axis=1) print(x_train.shape, y_train.shape) # 查看测试集, 去除 id X_test = test.drop(['id'], axis=1) print(X_test.shape) # 将测试集转换为适应 CNN 输入的 shape X_test = np.array(X_test).reshape(X_test.shape[0], X_test.shape[1], 1) print(X_test.shape, X_test.dtype)
3. 数据探索性分析
3.1 基本分析
train.head() # 查看前 5 条信息
test.head() # 查看前 5 条信息
train.describe()
test.describe()
train.info()
test.info()
综上可知：
- 主要的特征数据为 1 维信号振幅 (已被归一化至 0～1 了)，总长度均为 205 (205 个时间节点/心跳节拍)；
- 同时，除波形数据外，没有任何辅助或先验信息可以利用；
- 波形数据均已被量化为 float16 类型的数值型特征，且没有类别型特征需要考虑；
- 没有缺失值，无需填充，非常理想 —— 事实上，未采集到的信号默认振幅就是 0，故不存在缺失值的问题；
- 显然，这类非表格数据更适合用神经网络来处理，而非传统机器学习模型。
3.2 查看类别分布
plt.hist(train['label'], orientation = 'vertical', histtype = 'bar', color = 'red') plt.show()
可见，训练-验证集中存在 严重的类别不平衡问题。为此，有多种方式应对(https://blog.csdn.net/qq_39478403/article/details/115533295)。
试了很多种方法，最后发现还是用 SMOTE 对少数类别上采样效果最好：
# 使用 SMOTE 对数据进行上采样以解决类别不平衡问题 smote = SMOTE(random_state=2021, n_jobs=-1) k_x_train, k_y_train = smote.fit_resample(x_train, y_train) print(f"after smote, k_x_train.shape: {k_x_train.shape}, k_y_train.shape: {k_y_train.shape}") # 将训练集转换为适应 CNN 输入的 shape k_x_train = np.array(k_x_train).reshape(k_x_train.shape[0], k_x_train.shape[1], 1)
plt.hist(k_y_train, orientation = 'vertical', histtype = 'bar', color = 'blue') plt.show()
但这样做的缺点也显而易见 —— 运算量大幅度增加且容易导致过拟合，这也是我从 A 榜切换到 B 榜分数大跌的原因之一 T T
4. 辅助函数
# 评估函数 def abs_sum(y_pred, y_true): y_pred = np.array(y_pred) y_true = np.array(y_true) loss = sum(sum(abs(y_pred-y_true))) return loss
5. 模型训练与推理 (TensorFlow 2.2+)
- 由于数据特征形式较简单，仅 SMOTE 上采样处理后重复、相似性高，使用 过于复杂和深层 的神经网络将 极易过拟合，故本方案采用的模型都较为简单，且均使用了 Dropout 减轻过拟合(几乎是必须的操作)
- 共用 3 个单模融合，A 榜得分分别约为：Net1 142，Net3 156，Net8 145，此亦为加权融合设置权重的依据
- 模型融合并取得提升的前提是 好而不同，为此，3 个模型各自具有一定性能 (较好) 但在细节上都有一定差别 (不同)。模型的设计是本方案提升的关键，例如膨胀卷积、各种池化、分类器等，需细品 ^ ^
- 由于模型及其超参数在 B 榜时已确定和固定，不再划分训练集和验证集，直接使用全数据集训练模型 (但 A 榜平时采用 10-fold CV)
- 训练策略均为：学习率阶梯衰减策略 LearningRateScheduler + Adam 优化器 + sparse_categorical_crossentropy 损失函数 + batch_size 64 + epoch 30
5.1 Net 1
class Net1(K.Model): def __init__(self): super(Net1, self).__init__() self.conv1 = Conv1D(filters=16, kernel_size=3, padding='same', activation='relu', input_shape = (205, 1)) self.conv2 = Conv1D(filters=32, kernel_size=3, dilation_rate=2, padding='same', activation='relu') self.conv3 = Conv1D(filters=64, kernel_size=3, dilation_rate=2, padding='same', activation='relu') self.conv4 = Conv1D(filters=64, kernel_size=5, dilation_rate=2, padding='same', activation='relu') self.max_pool1 = MaxPool1D(pool_size=3, strides=2, padding='same') self.conv5 = Conv1D(filters=128, kernel_size=5, dilation_rate=2, padding='same', activation='relu') self.conv6 = Conv1D(filters=128, kernel_size=5, dilation_rate=2, padding='same', activation='relu') self.max_pool2 = MaxPool1D(pool_size=3, strides=2, padding='same') self.dropout = Dropout(0.5) self.flatten = Flatten() self.fc1 = Dense(units=256, activation='relu') self.fc21 = Dense(units=16, activation='relu') self.fc22 = Dense(units=256, activation='sigmoid') self.fc3 = Dense(units=4, activation='softmax') def call(self, x): x = self.conv1(x) x = self.conv2(x) x = self.conv3(x) x = self.conv4(x) x = self.max_pool1(x) x = self.conv5(x) x = self.conv6(x) x = self.max_pool2(x) x = self.dropout(x) x = self.flatten(x) x1 = self.fc1(x) x2 = self.fc22(self.fc21(x)) x = self.fc3(x1+x2) return x
5.2 Net 3
class GeMPooling(tf.keras.layers.Layer): def __init__(self, p=1.0, train_p=False): super().__init__() self.eps = 1e-6 self.p = tf.Variable(p, dtype=tf.float32) if train_p else p def call(self, inputs: tf.Tensor, **kwargs): inputs = tf.clip_by_value(inputs, clip_value_min=1e-6, clip_value_max=tf.reduce_max(inputs)) inputs = tf.pow(inputs, self.p) inputs = tf.reduce_mean(inputs, axis=[1], keepdims=False) inputs = tf.pow(inputs, 1./self.p) return inputs class Net3(K.Model): def __init__(self): super(Net3, self).__init__() self.conv1 = Conv1D(filters=16, kernel_size=3, padding='same', activation='relu', input_shape = (205, 1)) self.conv2 = Conv1D(filters=32, kernel_size=3, dilation_rate=2, padding='same', activation='relu') self.conv3 = Conv1D(filters=64, kernel_size=3, dilation_rate=2, padding='same', activation='relu') self.max_pool1 = MaxPool1D(pool_size=3, strides=2, padding='same') self.conv4 = Conv1D(filters=64, kernel_size=5, dilation_rate=2, padding='same', activation='relu') self.conv5 = Conv1D(filters=128, kernel_size=5, dilation_rate=2, padding='same', activation='relu') self.max_pool2 = MaxPool1D(pool_size=3, strides=2, padding='same') self.conv6 = Conv1D(filters=256, kernel_size=5, dilation_rate=2, padding='same', activation='relu') self.conv7 = Conv1D(filters=128, kernel_size=7, dilation_rate=2, padding='same', activation='relu') self.gempool = GeMPooling() self.dropout1 = Dropout(0.5) self.flatten = Flatten() self.fc1 = Dense(units=256, activation='relu') self.fc21 = Dense(units=16, activation='relu') self.fc22 = Dense(units=256, activation='sigmoid') self.fc3 = Dense(units=4, activation='softmax') def call(self, x): x = self.conv1(x) x = self.conv2(x) x = self.conv3(x) x = self.max_pool1(x) x = self.conv4(x) x = self.conv5(x) x = self.max_pool2(x) x = self.conv6(x) x = self.conv7(x) x = self.gempool(x) x = self.dropout1(x) x = self.flatten(x) x1 = self.fc1(x) x2 = self.fc22(self.fc21(x)) x = self.fc3(x1 + x2) return x
5.3 Net 8
class Net8(K.Model): def __init__(self): super(Net8, self).__init__() self.conv1 = Conv1D(filters=16, kernel_size=3, padding='same', activation='relu',input_shape = (205, 1)) self.conv2 = Conv1D(filters=32, kernel_size=3, padding='same', dilation_rate=2, activation='relu') self.conv3 = Conv1D(filters=64, kernel_size=3, padding='same', dilation_rate=2, activation='relu') self.conv4 = Conv1D(filters=128, kernel_size=3, padding='same', dilation_rate=2, activation='relu') self.conv5 = Conv1D(filters=128, kernel_size=5, padding='same', dilation_rate=2, activation='relu') self.max_pool1 = MaxPool1D(pool_size=3, strides=2, padding='same') self.avg_pool1 = AvgPool1D(pool_size=3, strides=2, padding='same') self.conv6 = Conv1D(filters=128, kernel_size=5, padding='same', dilation_rate=2, activation='relu') self.conv7 = Conv1D(filters=128, kernel_size=5, padding='same', dilation_rate=2, activation='relu') self.max_pool2 = MaxPool1D(pool_size=3, strides=2, padding='same') self.avg_pool2 = AvgPool1D(pool_size=3, strides=2, padding='same') self.dropout = Dropout(0.5) self.flatten = Flatten() self.fc1 = Dense(units=256, activation='relu') self.fc21 = Dense(units=16, activation='relu') self.fc22 = Dense(units=256, activation='sigmoid') self.fc3 = Dense(units=4, activation='softmax') def call(self, x): x = self.conv1(x) x = self.conv2(x) x = self.conv3(x) x = self.conv4(x) x = self.conv5(x) xm1 = self.max_pool1(x) xa1 = self.avg_pool1(x) x = tf.concat([xm1, xa1], 2) x = self.conv6(x) x = self.conv7(x) xm2 = self.max_pool2(x) xa2 = self.avg_pool2(x) x = tf.concat([xm2, xa2], 2) x = self.dropout(x) x = self.flatten(x) x1 = self.fc1(x) x2 = self.fc22(self.fc21(x)) x = self.fc3(x1+x2) return x
6. 软投票融合 + 阈值法
- 由于 3 个单模均输出类别概率预测值，故使用 Soft Voting 根据 A 榜单模得分设置权重，以实现预测结果加权融合
- 由于本赛题属于分类问题，可使用阈值法将预测概率不小于 0.5 的类别置 1，其余则置 0
- 对于预测概率均小于 0.5 的难样本，使用二次处理：若最大预测值比次大预测值至少高 0.04 (这个值需要自己把握)，则认为最大预测值足够可信并置 1 其余置 0；否则认为最大预测值和次大预测值区分度不够高，难以分辨不作处理，仅令最小的另外两个预测值置 0
# 根据 A 榜得分，加权融合预测结果 predictions_weighted = 0.35 * predictions_nn1 + 0.31 * predictions_nn3 + 0.34* predictions_nn8 predictions_weighted[:5]
# 准备提交结果 submit = pd.DataFrame() submit['id'] = range(100000, 120000) submit['label_0'] = predictions_weighted[:, 0] submit['label_1'] = predictions_weighted[:, 1] submit['label_2'] = predictions_weighted[:, 2] submit['label_3'] = predictions_weighted[:, 3] submit.head()
# 第一次后处理未涉及的难样本 index others = [] # 第一次后处理 - 将预测概率值大于 0.5 的样本的概率置 1，其余置 0 threshold = 0.5 for index, row in submit.iterrows(): row_max = max(list(row[1:])) # 当前行中的最大类别概率预测值 if row_max > threshold: for i in range(1, 5): if row[i] > threshold: submit.iloc[index, i] = 1 # 大于 0.5 的类别概率预测值置 1 else: submit.iloc[index, i] = 0 # 其余类别概率预测值置 0 else: others.append(index) # 否则，没有类别概率预测值不小于 0.5，加入第一次后处理未涉及的难样本列表，等待第二次后处理 print(index, row) submit.head(5)
# 第二次后处理 - 在预测概率值均不大于 0.5 的样本中，若最大预测值与次大预测值相差大于 0.04，则将最大预测值置 1，其余预测值置 0； # 否则，对最大预测值和次大预测值不处理 (难分类)，仅对其余样本预测值置 0 for idx in others: value = submit.iloc[idx].values[1:] ordered_value = sorted([(v, j) for j, v in enumerate(value)], reverse=True) # 根据类别概率预测值大小排序 #print(ordered_value) if ordered_value[0][0] - ordered_value[1][0] >= 0.04: # 最大与次大值相差至少 0.04 submit.iloc[idx, ordered_value[0][1]+1] = 1 # 则足够置信最大概率预测值并置为 1 for k in range(1, 4): submit.iloc[idx, ordered_value[k][1]+1] = 0 # 对非最大的其余三个类别概率预测值置 0 else: for s in range(2, 4): submit.iloc[idx, ordered_value[s][1]+1] = 0 # 难分样本，仅对最小的两个类别概率预测值置 0 print(submit.iloc[idx])
7. 保存结果用于提交
# 检视最后的预测结果 submit.head()
# 保存预测结果 submit.to_csv(("./submit_"+datetime.datetime.now().strftime('%Y%m%d_%H%M%S') + ".csv"), index=False)
8. 后话
- 对于这类非表格数据，我觉得显然不适合用 LGB、XGB 等。因为从我的角度看，每条样本的 205 个振幅实际是就是一个长度为 205 的特征，而非 205 个具有不同性质的特征，所以我一开始就采用神经网络。当然，LGB、XGB 也试过，效果差很多，在预料的范围之内
- 这不是十分严格的时间序列数据 (没有周期性)，我估摸着这仅仅是人的心跳的其中一个周期，所以针对时间序列预测的技术可能并不很好发挥
- 其实第 0 类样本数量最多，我估摸着这是正常人的心跳，而第 1、2、3 类应该是病人心跳。观察训练后的混淆矩阵容易知道，主要分类错误来自于将第 0 类错分为其他类 (误把正常人当病人)，这是一条潜在的改进线索
- 找到好用的数据增强方式并不容易，而其实 SMOTE 这种插值方式本身就是一种
- 提交结果之前先检查一下，不要一把梭就交上去了 ...
- 需要强大的毅力和耐心，起落沉浮非常考验心态 T T
--END--
关注阿里云天池，人人都可以玩转大数据
