arXiv reaDer
現場での不均一自己蒸留による貨物列車ブレーキシステムの効率的な視覚的故障検出
Efficient Visual Fault Detection for Freight Train Braking System via Heterogeneous Self Distillation in the Wild
貨物列車の効率的な視覚的故障検出は、制限されたハードウェア環境下で鉄道の安全な運行を確保するために重要な部分です。深層学習ベースのアプローチは物体検出には優れていますが、貨物列車の故障検出の効率は、現実世界のエンジニアリングに適用するにはまだ不十分です。この論文では、低リソース要件を満たしながら検出精度と速度を確保するための異種自己蒸留フレームワークを提案します。出力特徴ナレッジ内の特権情報は、パフォーマンスを向上させるために蒸留によって教師から生徒モデルに転送できます。まず、軽量のバックボーンを採用して特徴を抽出し、新しい異種知識のネックを生成します。このようなネックは、並列エンコードを通じて位置情報とチャネル間の長距離依存関係をモデル化し、特徴抽出機能を最適化します。次に、一般分布を利用して、より信頼性が高く正確な境界ボックス推定値を取得します。最後に、学習効率を向上させるためにネットワークがラベル付近の値に集中しやすくする新しい損失関数を採用します。 4 つの断層データセットの実験により、私たちのフレームワークが 1 秒あたり 37 フレーム以上を達成でき、従来の蒸留アプローチと比較して最高の精度を維持できることが明らかになりました。さらに、最先端の手法と比較して、当社のフレームワークはメモリ使用量が少なく、モデル サイズが最小であるため、より競争力のあるパフォーマンスを示します。
Efficient visual fault detection of freight trains is a critical part of ensuring the safe operation of railways under the restricted hardware environment. Although deep learning-based approaches have excelled in object detection, the efficiency of freight train fault detection is still insufficient to apply in real-world engineering. This paper proposes a heterogeneous self-distillation framework to ensure detection accuracy and speed while satisfying low resource requirements. The privileged information in the output feature knowledge can be transferred from the teacher to the student model through distillation to boost performance. We first adopt a lightweight backbone to extract features and generate a new heterogeneous knowledge neck. Such neck models positional information and long-range dependencies among channels through parallel encoding to optimize feature extraction capabilities. Then, we utilize the general distribution to obtain more credible and accurate bounding box estimates. Finally, we employ a novel loss function that makes the network easily concentrate on values near the label to improve learning efficiency. Experiments on four fault datasets reveal that our framework can achieve over 37 frames per second and maintain the highest accuracy in comparison with traditional distillation approaches. Moreover, compared to state-of-the-art methods, our framework demonstrates more competitive performance with lower memory usage and the smallest model size.
updated: Mon Jul 03 2023 01:27:39 GMT+0000 (UTC)
published: Mon Jul 03 2023 01:27:39 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アソシエイト