586. Child Mind Institute - Detect Sleep States | child-mind-institute-detect-sleep-states
首先,我要感谢Kaggle团队和主办方举办这次比赛,并感谢许多其他参与者分享代码、想法和数据集。特别感谢我的队友(furu、mizoo、isamu、tereka)一起在比赛中合作,以及感谢213tubo分享了如此棒的pipeline。
我们的最终提交是总共55个模型的集成,包括十一个5折模型。
我们的解决方案有四个主要部分。它们是furu、mizoo、kaerururu部分、isamu部分、tereka部分和后处理部分。
我们的pipeline基于213tubo的pipeline。
架构是各种特征提取器 -> Unet编码器 -> Unet1d解码器。
发作(onset)CV最佳模型和醒来(wakeup)CV最佳模型被分开保存。
特征工程
特征提取器
Unet编码器
Unet1d解码器
各种持续时间
Isamu有两种模型架构。一种是基于帕金森比赛第4名解决方案的MultiResidualBiGRU模型,另一种是基于Wavenet的模型。
Tereka有两种模型架构。一种是Unet基础模型,另一种是Transformer基础模型。
特征工程
使用模型输出的移动平均(窗口大小=12)。
| 名称 | CV | LB | Private |
|---|---|---|---|
| 提交1(CV最佳) | 0.8125 | 0.773 | 0.823 |
| 提交2(LB最佳) | - | 0.778 | 0.822 |
| 名称 | CV | LB | Private |
|---|---|---|---|
| 有后处理 | 0.8125 | 0.773 | 0.823(+0.011) |
| 无后处理 | - | 0.767 | 0.812 |
| 名称 | 模型 | CV | LB | Private |
|---|---|---|---|---|
| furu模型 | GRUINITFeatureExtractor-UNet1DDecoder | 0.7689665 | 0.746 | 0.8 |
| kaerururu模型1 | GRUFeatureExtractor-UNet1DDecoder | 0.743 | - | - |
| kaerururu模型2 | GRUFeatureExtractor-UNet1DDecoder | 0.733 | - | - |
| mizoo模型 | LSTMFeatureExtractor-UNet1DDecoder | 0.7508 | 0.736 | 0.784 |
| Isamu模型1 | wavenet_lstm | 0.773 | - | - |
| Isamu模型2 | 1d-gru | 0.775 | - | - |
| tereka模型1 | unet base | 0.764 | 0.752 | 0.803 |
| tereka模型2 | transformer base | 0.773 | 0.729 | 0.797 |