Skip to main navigation Skip to search Skip to main content

Robust Lightweight Depth Estimation Model via Data-Free Distillation

  • Zihan Gao
  • , Peng Gao
  • , Wei Yin
  • , Yifan Liu
  • , Zengchang Qin*
  • *Corresponding author for this work
  • Beihang University
  • Hangzhou Dianzi University
  • DJI Innovation Technology Co., Ltd.
  • Adelaide University

Research output: Contribution to journalConference articlepeer-review

Abstract

Existing Monocular Depth Estimation (MDE) methods often use large and complex neural networks. Despite the advanced performance of these methods, we consider the efficiency and generalization for practical applications with limited resources. In our paper, we present an efficient transformer-based monocular relative depth estimation network and train it with a diverse depth dataset to obtain good generalization performance. Knowledge distillation (KD) is employed to transfer the general knowledge from a pre-trained teacher network to the compact student network, demonstrating that KD can improve the generalization ability as well as the accuracy. Moreover, we propose a geometric label-free distillation method to improve the lightweight model in specific domains utilizing 3D geometric cues with unlabeled data. We show that our method outperforms other KD methods with or without ground truth supervision. Finally, we propose an application of the lightweight network to a two-stage depth completion task. Our method shows on par or even superior cross-domain generalization ability compared to large networks.

Original languageEnglish
Pages (from-to)4160-4164
Number of pages5
JournalProceedings - ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing
DOIs
StatePublished - 2024
Event2024 IEEE International Conference on Acoustics, Speech, and Signal Processing, ICASSP 2024 - Seoul, Korea, Republic of
Duration: 14 Apr 202419 Apr 2024

Keywords

  • Depth Completion
  • Depth Estimation
  • Generalization
  • Knowledge Distillation

Fingerprint

Dive into the research topics of 'Robust Lightweight Depth Estimation Model via Data-Free Distillation'. Together they form a unique fingerprint.

Cite this