arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2411.02149cs.CV

通过稳定对抗训练提升自监督单目深度估计的域泛化能力

Improving Domain Generalization in Self-supervised Monocular Depth Estimation via Stabilized Adversarial Training

Faculty of Computing, Harbin Institute of Technology(哈尔滨工业大学计算机学院);Academy for Engineering & Technology, Fudan University(复旦大学工程与技术学院)

Yuanqi Yao, Gang Wu, Kui Jiang, Siao Liu, Jian Kuai, Xianming Liu, Junjun Jiang

AI总结:

针对自监督单目深度估计泛化难题,提出稳定冲突优化对抗训练框架(SCAT),通过缩放深度网络与冲突梯度手术策略,在五个基准上实现最先进性能并显著提升泛化能力。

摘要:

学习一个具有良好泛化能力的自监督单目深度估计(MDE)模型仍然极具挑战性。尽管对抗增强在监督学习泛化中取得了成功,但将其简单纳入自监督MDE模型可能导致过度正则化,从而遭受严重的性能退化。在本文中,我们进行了定性分析并阐明了主要原因:(i)类似UNet的深度网络固有的敏感性,以及(ii)由过度正则化引起的双重优化冲突。为了解决这些问题,我们提出了一个通用的对抗训练框架,名为稳定冲突优化对抗训练(SCAT),将对抗数据增强整合到自监督MDE方法中,以实现稳定性与泛化之间的平衡。具体来说,我们设计了一个有效的缩放深度网络,调整长跳跃连接的系数,并有效稳定训练过程。然后,我们提出了一种冲突梯度手术策略,该策略逐步整合对抗梯度,并朝着无冲突方向优化模型。在五个基准上的大量实验表明,SCAT能够实现最先进的性能,并显著提高现有自监督MDE方法的泛化能力。

英文摘要:

Learning a self-supervised Monocular Depth Estimation (MDE) model with great generalization remains significantly challenging. Despite the success of adversarial augmentation in the supervised learning generalization, naively incorporating it into self-supervised MDE models potentially causes over-regularization, suffering from severe performance degradation. In this paper, we conduct qualitative analysis and illuminate the main causes: (i) inherent sensitivity in the UNet-alike depth network and (ii) dual optimization conflict caused by over-regularization. To tackle these issues, we propose a general adversarial training framework, named Stabilized Conflict-optimization Adversarial Training (SCAT), integrating adversarial data augmentation into self-supervised MDE methods to achieve a balance between stability and generalization. Specifically, we devise an effective scaling depth network that tunes the coefficients of long skip connection and effectively stabilizes the training process. Then, we propose a conflict gradient surgery strategy, which progressively integrates the adversarial gradient and optimizes the model toward a conflict-free direction. Extensive experiments on five benchmarks demonstrate that SCAT can achieve state-of-the-art performance and significantly improve the generalization capability of existing self-supervised MDE methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑