Distilling Temporal Knowledge with Masked Feature Reconstruction for 3D Object Detection

Haowen Zheng,Dong Cao,Jintao Xu,Rui Ai,Weihao Gu,Yang Yang,Yanyan Liang
2024-01-08
Abstract:Striking a balance between precision and efficiency presents a prominent challenge in the bird's-eye-view (BEV) 3D object detection. Although previous camera-based BEV methods achieved remarkable performance by incorporating long-term temporal information, most of them still face the problem of low efficiency. One potential solution is knowledge distillation. Existing distillation methods only focus on reconstructing spatial features, while overlooking temporal knowledge. To this end, we propose TempDistiller, a Temporal knowledge Distiller, to acquire long-term memory from a teacher detector when provided with a limited number of frames. Specifically, a reconstruction target is formulated by integrating long-term temporal knowledge through self-attention operation applied to feature teachers. Subsequently, novel features are generated for masked student features via a generator. Ultimately, we utilize this reconstruction target to reconstruct the student features. In addition, we also explore temporal relational knowledge when inputting full frames for the student model. We verify the effectiveness of the proposed method on the nuScenes benchmark. The experimental results show our method obtain an enhancement of +1.6 mAP and +1.1 NDS compared to the baseline, a speed improvement of approximately 6 FPS after compressing temporal knowledge, and the most accurate velocity estimation.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper attempts to solve the problem of balancing accuracy and efficiency in bird - eye - view (BEV) 3D object detection. Although previous camera - based BEV methods have achieved significant performance improvements through fusing long - term temporal information, most methods still face the problem of inefficiency. The paper proposes a temporal knowledge distillation method named TempDistiller, which aims to obtain long - term memory from the teacher detector, and this can be achieved even when a limited number of frames are provided. Specifically, the long - term temporal knowledge in the teacher features is integrated through self - attention operations to form a reconstruction target, and new features are generated for the masked parts in the student features by the generator, and finally the student features are reconstructed according to the reconstruction target. In addition, when the complete frames are input, the paper also explores the temporal relationship knowledge to further improve the speed and accuracy, especially in speed estimation. The main contributions of the paper include: - Proposing the first method to transfer temporal knowledge in 3D object detection, and exploring how to learn temporal knowledge and its relationship information. - Providing a novel perspective. Through knowledge distillation to handle long - term temporal fusion, the student detector can obtain long - term temporal knowledge from the teacher detector even when the number of input frames is reduced. - The experimental results show that compared with the baseline, this method improves 1.6 mAP and 1.1 NDS, the inference speed is increased by about 6 FPS, and at the same time, the most accurate speed estimation is achieved. These contributions not only solve the efficiency problems existing in the existing methods when dealing with long - term temporal information, but also provide a new direction for future 3D object detection research.