用多模态传感器数据自监督训练机器人感知模型,提升复杂环境下的鲁棒性。
OctoSense: Self-Supervised Learning for Multimodal Robot Perception
- 采用晚期融合掩码自编码器,针对不同传感器特性设计专用编码器
- 在夜间或传感器失效时仍保持稳定表现,推理速度达6.68毫秒
- 适合自动驾驶与四足机器人等真实场景的多模态感知任务
我们提出OctoSense,一个开源的多传感器平台,包含双目RGB相机、事件相机、激光雷达、热成像仪、惯性测量单元、带RTK校正的全球定位系统以及车辆CAN总线数据和四足机器人关节角度信息。同名的OctoSense数据集包含59小时的时间同步驾驶数据,涵盖不同环境、不同时段,包括传感器严重退化的场景。我们展示了利用此类真实机器人数据进行多模态自监督学习的方法:各传感器具有不同的表示形式、采样频率、延迟和噪声。所提方法为“晚期融合”掩码自编码器,(i) 使用模态专用分词器处理不同传感器的时空特性,(ii) 在推理时缓存各模态的特征令牌,实现对新输入的实时处理。该架构 (i) 推理速度快(在NVIDIA 5090上仅需6.68毫秒,Orin NX上为112毫秒),(ii) 在光流估计、深度预测、语义分割和自运动估计(平移、旋转、转向角)等任务上优于现有图像单模态基础模型,(iii) 在夜间或传感数据受损情况下仍具备强鲁棒性。项目页面提供数据集、代码及补充视频链接:https://abisulco.com/octosense/
原文摘要 · Abstract (English)
We present OctoSense, an open-source sensor platform with stereo RGB and event cameras, LiDAR, a thermal camera, an inertial measurement unit, RTK-corrected global positioning system, and proprioception (CAN bus data from a car, and joint angles for a quadruped robot). The eponymous OctoSense dataset contains 59 hours of time-synchronized driving data across different types of environments at different times of the day, including situations with highly degraded sensors. We demonstrate multi-modal self-supervised learning using such real-world robotics data, where sensors have different representations, frequencies, latencies and noise. Our approach, a "late-fusion" masked autoencoder, (i) uses modality-specific tokenizers to account for different spatiotemporal characteristics of these sensors, and (ii) caches modality-specific tokens at inference time to process new measurements as they come. This architecture (i) is fast (6.68 ms and 112 ms on NVIDIA 5090 and Orin NX respectively, to compute the representation), (ii) performs better than existing image-only foundation models on tasks such as estimation of optical flow, depth, semantic segmentation, and ego-motion (translation, rotation, and steering angle), and (iii) predicts robustly at nighttime or in situations where sensory data is degraded. See our project page for links to the dataset, code, and supplementary videos: https://abisulco.com/octosense/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。