# Risk Stratification of Lung Nodules Using 3D CNN-Based Multi-task Learning

arXiv: https://arxiv.org/abs/1704.08797 · HTML: https://mlsanity.com/paper/1704.08797/

## Key facts

- Cited by 14 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 28 Apr 2017.

## Built on by (most cited first, 14 of 14)

- [MedKLIP: Medical Knowledge Enhanced Language-Image Pre-Training in Radiology](https://mlsanity.com/paper/2301.02228/) (2023)
- [Knowledge-enhanced Visual-Language Pre-training on Chest Radiology Images](https://mlsanity.com/paper/2302.14042/) (2023)
- [Diagnose Like a Radiologist: Hybrid Neuro-Probabilistic Reasoning for Attribute-Based Medical Image Diagnosis](https://mlsanity.com/paper/2208.09282/) (2022)
- [UniBrain: Universal Brain MRI Diagnosis with Hierarchical Knowledge-enhanced Pre-training](https://mlsanity.com/paper/2309.06828/) (2023)
- [Interpretable Medical Image Classification using Prototype Learning and Privileged Information](https://mlsanity.com/paper/2310.15741/) (2023)
- [Autoregressive Sequence Modeling for 3D Medical Image Representation](https://mlsanity.com/paper/2409.08691/) (2024)
- [DeepAnalyze: Learning to Localize Crashes at Scale](https://mlsanity.com/paper/2109.14326/) (2021)
- [Re-thinking and Re-labeling LIDC-IDRI for Robust Pulmonary Cancer Prediction](https://mlsanity.com/paper/2207.14238/) (2022)
- [The Past, Current, and Future of Neonatal Intensive Care Units with Artificial Intelligence](https://mlsanity.com/paper/2302.00225/) (2023)
- [Plot2API: Recommending Graphic API from Plot via Semantic Parsing Guided Neural Network](https://mlsanity.com/paper/2104.01032/) (2021)
- [Interpretative Computer-aided Lung Cancer Diagnosis: from Radiology Analysis to Malignancy Evaluation](https://mlsanity.com/paper/2102.10919/) (2021)
- [Faithful learning with sure data for lung nodule diagnosis](https://mlsanity.com/paper/2202.12515/) (2022)
- [Machine Learning Applications in Lung Cancer Diagnosis, Treatment and Prognosis](https://mlsanity.com/paper/2203.02794/) (2022)
- [ViKL: A Mammography Interpretation Framework via Multimodal Aggregation of Visual-knowledge-linguistic Features](https://mlsanity.com/paper/2409.15744/) (2024)

## Abstract (arXiv)

Risk stratification of lung nodules is a task of primary importance in lung cancer diagnosis. Any improvement in robust and accurate nodule characterization can assist in identifying cancer stage, prognosis, and improving treatment planning. In this study, we propose a 3D Convolutional Neural Network (CNN) based nodule characterization strategy. With a completely 3D approach, we utilize the volumetric information from a CT scan which would be otherwise lost in the conventional 2D CNN based approaches. In order to address the need for a large amount for training data for CNN, we resort to transfer learning to obtain highly discriminative features. Moreover, we also acquire the task dependent feature representation for six high-level nodule attributes and fuse this complementary information via a Multi-task learning (MTL) framework. Finally, we propose to incorporate potential disagreement among radiologists while scoring different nodule attributes in a graph regularized sparse multi-task learning. We evaluated our proposed approach on one of the largest publicly available lung nodule datasets comprising 1018 scans and obtained state-of-the-art results in regressing the malignancy scores.