自動肺結節分類は、肺がんの早期診断にとって重要です。最近、深層学習技術により、この分野で目覚ましい進歩が見られました。ただし、これらの深いモデルは通常、計算が非常に複雑であり、ブラックボックス方式で機能します。これらの課題に対処するために、この作業では、効率的で(部分的に)説明可能な分類モデルを構築することを目指しています。特に、ニューラルアーキテクチャ検索(NAS)を使用して、優れた精度と速度のトレードオフで3Dネットワークアーキテクチャを自動的に検索します。さらに、ネットワークで畳み込みブロック注意モジュール(CBAM)を使用します。これは、推論プロセスを理解するのに役立ちます。トレーニング中、A-Softmax損失を使用して、角度的に識別可能な表現を学習します。推論段階では、多様なニューラルネットワークのアンサンブルを使用して、予測の精度と堅牢性を向上させます。 LIDC-IDRIデータベースで広範な実験を行っています。以前の最先端技術と比較して、私たちのモデルは、1/40未満のパラメーターを使用することにより、非常に匹敵するパフォーマンスを示しています。その上、実証的研究は、学習されたネットワークの推論プロセスが医師の診断と一致していることを示しています。関連するコードと結果は、https://github.com/fei-hdu/NAS-Lungでリリースされています。
Automatic pulmonary nodules classification is significant for early diagnosis of lung cancers. Recently, deep learning techniques have enabled remarkable progress in this field. However, these deep models are typically of high computational complexity and work in a black-box manner. To combat these challenges, in this work, we aim to build an efficient and (partially) explainable classification model. Specially, we use neural architecture search (NAS) to automatically search 3D network architectures with excellent accuracy/speed trade-off. Besides, we use the convolutional block attention module (CBAM) in the networks, which helps us understand the reasoning process. During training, we use A-Softmax loss to learn angularly discriminative representations. In the inference stage, we employ an ensemble of diverse neural networks to improve the prediction accuracy and robustness. We conduct extensive experiments on the LIDC-IDRI database. Compared with previous state-of-the-art, our model shows highly comparable performance by using less than 1/40 parameters. Besides, empirical study shows that the reasoning process of learned networks is in conformity with physicians' diagnosis. Related code and results have been released at: https://github.com/fei-hdu/NAS-Lung.