MiniVLN: Efficient Vision-and-Language Navigation by Progressive Knowledge Distillation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhu, Junyou, Qiao, Yanyuan, Zhang, Siqi, He, Xingjian, Wu, Qi, Liu, Jing
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912048407904256
author Zhu, Junyou
Qiao, Yanyuan
Zhang, Siqi
He, Xingjian
Wu, Qi
Liu, Jing
author_facet Zhu, Junyou
Qiao, Yanyuan
Zhang, Siqi
He, Xingjian
Wu, Qi
Liu, Jing
contents In recent years, Embodied Artificial Intelligence (Embodied AI) has advanced rapidly, yet the increasing size of models conflicts with the limited computational capabilities of Embodied AI platforms. To address this challenge, we aim to achieve both high model performance and practical deployability. Specifically, we focus on Vision-and-Language Navigation (VLN), a core task in Embodied AI. This paper introduces a two-stage knowledge distillation framework, producing a student model, MiniVLN, and showcasing the significant potential of distillation techniques in developing lightweight models. The proposed method aims to capture fine-grained knowledge during the pretraining phase and navigation-specific knowledge during the fine-tuning phase. Our findings indicate that the two-stage distillation approach is more effective in narrowing the performance gap between the teacher model and the student model compared to single-stage distillation. On the public R2R and REVERIE benchmarks, MiniVLN achieves performance on par with the teacher model while having only about 12% of the teacher model's parameter count.
format Preprint
id arxiv_https___arxiv_org_abs_2409_18800
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MiniVLN: Efficient Vision-and-Language Navigation by Progressive Knowledge Distillation
Zhu, Junyou
Qiao, Yanyuan
Zhang, Siqi
He, Xingjian
Wu, Qi
Liu, Jing
Computer Vision and Pattern Recognition
In recent years, Embodied Artificial Intelligence (Embodied AI) has advanced rapidly, yet the increasing size of models conflicts with the limited computational capabilities of Embodied AI platforms. To address this challenge, we aim to achieve both high model performance and practical deployability. Specifically, we focus on Vision-and-Language Navigation (VLN), a core task in Embodied AI. This paper introduces a two-stage knowledge distillation framework, producing a student model, MiniVLN, and showcasing the significant potential of distillation techniques in developing lightweight models. The proposed method aims to capture fine-grained knowledge during the pretraining phase and navigation-specific knowledge during the fine-tuning phase. Our findings indicate that the two-stage distillation approach is more effective in narrowing the performance gap between the teacher model and the student model compared to single-stage distillation. On the public R2R and REVERIE benchmarks, MiniVLN achieves performance on par with the teacher model while having only about 12% of the teacher model's parameter count.
title MiniVLN: Efficient Vision-and-Language Navigation by Progressive Knowledge Distillation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.18800