This repository is the official implementation of ACL for the binary classification task in Thyroid Nodule Diagnosis. In this paper, we propose a adaptive curriculum learning based method, ACL, to tackle with the inconsistent label problem. Specifically, ACL adaptively discovers and discards the samples with inconsistent labels via an adaptive threshold, calculated by the hard sample queue and the certainty queue.

Moreover, we contribute TNCD: a Thyroid Nodule Classification Dataset to facilitate future related research on the thyroid nodules. Extensive experimental results on TNCD based on three different backbone networks not only demonstrate the superiority of our method but also prove that the less-is-more principle which strategically discards the samples with Inconsistent Label could yield performance gains.

Please cite this paper in your publications if it helps your research

@inproceedings{gong2022less,
  title={Less is More: Adaptive Curriculum Learning for Thyroid Nodule Diagnosis},
  author={Gong, Haifan and Cheng, Hui and Xie, Yifan and Tan, Shuangyi and Chen, Guanqi and Chen, Fei and Li, Guanbin},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={248--257},
  year={2022},
  organization={Springer}
}

CUDA 11.2

torch 1.11

torchvision 0.12

• Clone this repo:

git clone https://github.com/chenghui-666/ACL/

• Install necessary packages using pip install -r requirements.txt or conda env create -f environment.yaml -n <your_env_name>.

The processed data should be downloaded via Baidu Drive with the extract code: eehb. The trained models are available at Baidu Drive with extract code: dp1w. Or you can download the data from Google Drive.

The downloaded file should be extracted to ./datasets/ directory.

1. Our method:

   sh acl_train.sh
   sh acl_inference.sh

2. Baselines:

   sh baseline.sh

3. Ablation study:

   sh ablation.sh

• The result csv file can be found in the file ./results.csv.

MIT License

haifangong@outlook.com

chenghui_123@foxmail.com