技术深度解析
Inertia-1的核心创新并非在于一种全新的架构,而在于一种精心设计的预训练策略,该策略欣然接纳了真实世界传感器数据的混乱特性。该模型基于改进的时序Transformer(TST)架构构建,其精神与Google的Perceiver IO中使用的TST类似,但针对IMU数据的独特特性进行了优化。
架构细节:
- 输入处理: 模型接受原始的6轴IMU数据(3轴加速度计 + 3轴陀螺仪)作为时间步序列。关键在于,它使用了一种可学习的位置编码,该编码基于实际时间戳而非序列索引。这使其能够原生处理从20Hz到200Hz的可变采样率。
- 分块嵌入: 输入序列被划分为不重叠的块,每块包含50个时间步。每个块通过一个小型MLP被投影到一个512维的嵌入向量中。这降低了序列长度,同时保留了局部时间结构。
- Transformer编码器: 一个由12个Transformer层组成的堆叠,每层有8个注意力头。注意力机制经过修改,包含了一个相对位置偏置,这有助于模型通过学习传感器轴之间的空间关系,泛化到不同的传感器佩戴位置(例如,手腕 vs. 脚踝)。
- 预训练目标: 模型使用掩码自编码(MAE)目标进行预训练。输入信号的随机块(通常为40%)被掩码,模型必须重建缺失的传感器读数。这迫使模型学习一种丰富的、上下文相关的人体运动动力学理解。
训练数据: 预训练语料库是关键的区别所在。研究团队从15个不同的公开数据集中组装了一个超过10,000小时的带标签和未带标签的IMU数据集,包括:
- MotionSense: 来自24名受试者执行6种活动的智能手机加速度计数据。
- WISDM: 来自51名受试者执行18种活动的腕戴式加速度计数据。
- RealWorld HAR: 来自15名受试者的数据,传感器放置在胸部、前臂、头部、小腿和腰部,执行8种活动。
- PAMAP2: 9名受试者佩戴3个IMU传感器(手腕、胸部、脚踝)执行18种活动。
- 一个自定义数据集: 从50名志愿者在日常生活中的佩戴原型腕带收集的500小时未带标签数据。
基准性能: 研究团队将Inertia-1与多个最先进的任务特定模型以及一个通用时序基础模型(TimesFM)进行了评估。结果令人瞩目:
| 模型 | HAR70+(零样本F1) | RealWorld HAR(零样本F1) | PAMAP2(微调,10%数据) | 推理延迟(每10秒窗口,毫秒) |
|---|---|---|---|---|
| Inertia-1 (Base) | 0.82 | 0.79 | 0.94 | 12.3 |
| Inertia-1 (Large) | 0.85 | 0.82 | 0.96 | 28.7 |
| DeepConvLSTM(任务特定SOTA) | 0.76 | 0.74 | 0.91 | 8.1 |
| TimesFM(通用时序基础模型) | 0.68 | 0.65 | 0.85 | 15.2 |
| 随机森林(手工特征) | 0.61 | 0.58 | 0.79 | 2.4 |
数据要点: 在零样本设置下,Inertia-1比此前最先进的任务特定模型实现了6-8%的绝对F1分数提升,展示了真正的泛化能力。微调效率同样令人印象深刻:仅使用PAMAP2训练数据的10%,它就超越了使用全部数据训练的DeepConvLSTM的性能。这表明,预训练表示捕获了基本的运动基元,这些基元能够有效地跨活动和传感器配置进行迁移。
该模型在GitHub上以仓库`inertia-motion/inertia-1`提供,发布首周已获得超过2800颗星。该仓库包含完整的训练代码、Base(1.1亿参数)和Large(3.4亿参数)两种变体的模型权重,以及一个用于处理原始IMU数据的详细数据管道。
关键参与者与案例研究
Inertia-1项目是由Dr. Anya Sharma(前MIT媒体实验室成员)和Dr. Kenji Tanaka(东京大学)领导的一个合作团队的智慧结晶。他们组建了一个来自卡内基梅隆大学、苏黎世联邦理工学院和清华大学的研究人员联盟。该项目部分由美国国家科学基金会网络-物理系统项目提供的320万美元资助。
竞争方法: 可穿戴AI领域目前由两种范式主导:
1. 任务特定模型: 像Apple(watchOS中的活动分类)、Fitbit(睡眠阶段检测)和Garmin(跑步动力学)这样的公司,会针对其自有硬件上的特定任务训练高度专业化的模型。这些模型针对单一的传感器佩戴位置和采样率进行了优化,实现了高精度,但零可迁移性。
2. 通用时序基础模型: Google的TimesFM和Amazon的Chronos是通用时序基础模型,在大量时序数据(金融、天气、能源)上训练。虽然强大,但它们并未针对人类运动的独特特性进行优化。