arXiv reaDer
ビジョンベースのレーンセンタリングのための信頼ベースのフェデレーション蒸留
Confidence-based federated distillation for vision-based lane-centering
自動運転の基本的な課題は、ステアリング角度を調整して車両を車線の中央に維持することです。最近の進歩では、ディープ ニューラル ネットワークを活用して、車のカメラで撮影した画像から直接ステアリングの決定を予測します。機械学習ベースのステアリング角予測では、モデルのトレーニングのために大量の潜在的なプライベート データをアップロードする際の車両の制限を考慮する必要があります。フェデレーテッド ラーニングは、プライベート データを共有せずに複数の車両がグローバル モデルを共同でトレーニングできるようにすることで、これらの制約に対処できますが、データの分布が車両間で非 iid であることが多いため、良好な精度を達成するのは困難です。この論文では、舵角予測のための連合学習のパフォーマンスを向上させるための、新しい信頼ベースの連合蒸留手法を紹介します。具体的には、各ローカル モデルの予測信頼性を決定するためのエントロピーの新しい使用法を提案し、次に、グローバル モデルの学習をガイドする教師として最も信頼性の高いローカル モデルを選択します。ビジョンベースの車線センタリングの包括的な評価では、提案されたアプローチが FedAvg と FedDF をそれぞれ 11.3% と 9% 上回るパフォーマンスを発揮できることが示されています。
A fundamental challenge of autonomous driving is maintaining the vehicle in the center of the lane by adjusting the steering angle. Recent advances leverage deep neural networks to predict steering decisions directly from images captured by the car cameras. Machine learning-based steering angle prediction needs to consider the vehicle's limitation in uploading large amounts of potentially private data for model training. Federated learning can address these constraints by enabling multiple vehicles to collaboratively train a global model without sharing their private data, but it is difficult to achieve good accuracy as the data distribution is often non-i.i.d. across the vehicles. This paper presents a new confidence-based federated distillation method to improve the performance of federated learning for steering angle prediction. Specifically, it proposes the novel use of entropy to determine the predictive confidence of each local model, and then selects the most confident local model as the teacher to guide the learning of the global model. A comprehensive evaluation of vision-based lane centering shows that the proposed approach can outperform FedAvg and FedDF by 11.3% and 9%, respectively.
updated: Mon Jun 05 2023 20:16:19 GMT+0000 (UTC)
published: Mon Jun 05 2023 20:16:19 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト