Knowledge Distillation for Road Detection based on cross-model Semi-Supervised Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ma, Wanli, Karakus, Oktay, Rosin, Paul L.
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909149063806976
author Ma, Wanli
Karakus, Oktay
Rosin, Paul L.
author_facet Ma, Wanli
Karakus, Oktay
Rosin, Paul L.
contents The advancement of knowledge distillation has played a crucial role in enabling the transfer of knowledge from larger teacher models to smaller and more efficient student models, and is particularly beneficial for online and resource-constrained applications. The effectiveness of the student model heavily relies on the quality of the distilled knowledge received from the teacher. Given the accessibility of unlabelled remote sensing data, semi-supervised learning has become a prevalent strategy for enhancing model performance. However, relying solely on semi-supervised learning with smaller models may be insufficient due to their limited capacity for feature extraction. This limitation restricts their ability to exploit training data. To address this issue, we propose an integrated approach that combines knowledge distillation and semi-supervised learning methods. This hybrid approach leverages the robust capabilities of large models to effectively utilise large unlabelled data whilst subsequently providing the small student model with rich and informative features for enhancement. The proposed semi-supervised learning-based knowledge distillation (SSLKD) approach demonstrates a notable improvement in the performance of the student model, in the application of road segmentation, surpassing the effectiveness of traditional semi-supervised learning methods.
format Preprint
id arxiv_https___arxiv_org_abs_2402_05305
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Knowledge Distillation for Road Detection based on cross-model Semi-Supervised Learning
Ma, Wanli
Karakus, Oktay
Rosin, Paul L.
Computer Vision and Pattern Recognition
The advancement of knowledge distillation has played a crucial role in enabling the transfer of knowledge from larger teacher models to smaller and more efficient student models, and is particularly beneficial for online and resource-constrained applications. The effectiveness of the student model heavily relies on the quality of the distilled knowledge received from the teacher. Given the accessibility of unlabelled remote sensing data, semi-supervised learning has become a prevalent strategy for enhancing model performance. However, relying solely on semi-supervised learning with smaller models may be insufficient due to their limited capacity for feature extraction. This limitation restricts their ability to exploit training data. To address this issue, we propose an integrated approach that combines knowledge distillation and semi-supervised learning methods. This hybrid approach leverages the robust capabilities of large models to effectively utilise large unlabelled data whilst subsequently providing the small student model with rich and informative features for enhancement. The proposed semi-supervised learning-based knowledge distillation (SSLKD) approach demonstrates a notable improvement in the performance of the student model, in the application of road segmentation, surpassing the effectiveness of traditional semi-supervised learning methods.
title Knowledge Distillation for Road Detection based on cross-model Semi-Supervised Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2402.05305