用智能课程学习自动调参,让视觉SLAM训练快10倍
AutoLoop: Fast Visual SLAM Fine-tuning through Agentic Curriculum Learning
- 用强化学习代理动态调整回环权重,免去手动调参
- 在TartanAir训练,跨多个数据集性能相当或更好
- 适合需要快速适配新场景的SLAM系统开发者
当前视觉SLAM系统在计算效率与回环检测鲁棒性之间难以平衡。传统方法需人工精细调参且开销大,基于学习的方法或缺乏显式回环能力,或依赖高耗能实现。我们提出AutoLoop,结合自动化课程学习与高效微调,采用DDPG(深度确定性策略梯度)代理在训练中动态调整回环权重,避免手动超参数搜索,显著减少训练步数。该方法离线预计算潜在回环匹配对,并通过代理引导的课程机制实现模型对新场景的高效适应。在TartanAir上训练并经KITTI、EuRoC、ICL-NUIM和TUM RGB-D多基准测试验证,AutoLoop在保持或提升性能的同时,训练时间相比传统方法减少一个数量级。结果表明,这种自动化课程策略不仅加速训练,还能在多种环境条件下维持或增强模型表现。
原文摘要 · Abstract (English)
Current visual SLAM systems face significant challenges in balancing computational efficiency with robust loop closure handling. Traditional approaches require careful manual tuning and incur substantial computational overhead, while learning-based methods either lack explicit loop closure capabilities or implement them through computationally expensive methods. We present AutoLoop, a novel approach that combines automated curriculum learning with efficient fine-tuning for visual SLAM systems. Our method employs a DDPG (Deep Deterministic Policy Gradient) agent to dynamically adjust loop closure weights during training, eliminating the need for manual hyperparameter search while significantly reducing the required training steps. The approach pre-computes potential loop closure pairs offline and leverages them through an agent-guided curriculum, allowing the model to adapt efficiently to new scenarios. Experiments conducted on TartanAir for training and validated across multiple benchmarks including KITTI, EuRoC, ICL-NUIM and TUM RGB-D demonstrate that AutoLoop achieves comparable or superior performance while reducing training time by an order of magnitude compared to traditional approaches. AutoLoop provides a practical solution for rapid adaptation of visual SLAM systems, automating the weight tuning process that traditionally requires multiple manual iterations. Our results show that this automated curriculum strategy not only accelerates training but also maintains or improves the model's performance across diverse environmental conditions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。