arXiv:2607.01851cs.CV2026-07中稿 · ECCV

用知识蒸馏压缩大模型,让月球3D重建在低算力设备上高效运行

Geometric Foundation Model Distillation for Efficient Lunar 3D Reconstruction

论文配图:Geometric Foundation Model Distillation for Efficient Lunar 3D Reconstruction
图 1 · 摘自论文原文
  • 通过SVD结构化初始化,将大模型几何知识迁移到小模型
  • 模型体积缩小7倍,仍保持接近原模型的重建精度
  • 适合资源受限场景下的3D视觉任务部署

大型3D基础模型如MASt3R虽能实现顶尖的立体重建效果,但计算开销大,在行星探测等对机载算力严格受限的领域难以部署。本文以月球立体重建为挑战性且实用的案例,研究如何通过知识蒸馏压缩此类模型。基于一个在月球图像上微调的688M参数MASt3R教师模型,我们将其密集几何预测蒸馏到一系列轻量级学生模型中,涵盖不同编码器类型(CNN与ViT)、解码器宽度与深度及训练策略。为解决师生模型间的维度不匹配问题,提出一种基于SVD的结构化初始化方法,将教师解码器权重投影至学生更小的隐空间,提供良好起点,显著提升收敛速度与最终性能。在月球数据上的实验表明,可获得一个模型规模缩小达7倍、仍保留大部分教师模型重建准确率的学生模型,甚至优于仅用稀疏真值标注训练的基线。研究还揭示:卷积编码器表现弱于基于Transformer的替代方案(尽管预训练可用性是混杂因素),保留编码器容量比扩大解码器更重要,特征级蒸馏始终优于仅输出监督,且基于SVD的初始化可提升优化稳定性。这些发现为在资源受限环境中部署3D重建模型提供了实用指导。

原文摘要 · Abstract (English)

Large 3D foundation models such as MASt3R achieve state-of-the-art stereo reconstruction but are computationally demanding for deployment under strict hardware constraints -- a critical limitation in domains such as planetary exploration, where onboard computing is severely restricted. We study how far such models can be compressed through knowledge distillation, using lunar stereo reconstruction as a challenging and practically relevant case study. Starting from a 688M-parameter MASt3R teacher fine-tuned on lunar imagery, we distill its dense geometric predictions into a family of lightweight students spanning different encoder types (CNN vs ViT), decoder widths and depths, and training strategies. To bridge the dimensional mismatch between teacher and student, we propose a structured SVD-based initialization that projects the teacher's decoder weights into the student's smaller latent space, yielding a warm start that significantly improves convergence and final performance. Based on our results on lunar data, we can obtain a distilled student that retains most of teacher's reconstruction accuracy while reducing the model size up to 7 times, and even outperforms a baseline trained directly with sparse ground-truth annotations. Beyond compression, our study highlights both principles and practical insights for distilling geometric foundation models: a convolutional encoder underperforms transformer-based alternatives (though pretraining availability remains a confounding factor), preserving encoder capacity is more critical than maintaining a large decoder, feature-level distillation consistently outperforms output-only supervision, and SVD-based initialization improves optimisation stability. These findings provide practical guidelines for deploying 3D reconstruction models in resource-constrained environments.

3D重建知识蒸馏轻量化月球探测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。