BoMD: Bag of Multi-label Descriptors for Noisy Chest X-ray Classification

Official code for "BoMD: Bag of Multi-label Descriptors for Noisy Chest X-ray Classification"

BoMD: Bag of Multi-label Descriptors for Noisy Chest X-ray Classification,
Yuanhong Chen*, Fengbei Liu*, Hu Wang, Chong Wang, Yu Tian, Yuyuan Liu, Gustavo Carneiro.
ICCV 2023 (arXiv 2203.01937)

Download links

Weights

• Word embeddings extracted by the BLUEBert model.
• Classifier weights trained on NIH
• MID & NSD weights for run-20220816_175557-6dn2k0id

Datasets

Download the raw data: ChestXpert, NIH, PadChest, OpenI, NIH-GOOGLE

Download the processed OpenI & PadChest

Training

Train BoMD on NIH with the following command:

bash run_train_bomd_nih.sh

Train classificaiton head on NIH with the MID and NSD checkpoints:

bash run_train_bomd_nih_cls

Testing

bash run_eval_bomd_nih.sh

Reference

@article{chen2022bomd,
  title={BoMD: Bag of Multi-label Descriptors for Noisy Chest X-ray Classification},
  author={Chen, Yuanhong and Liu, Fengbei and Wang, Hu and Wang, Chong and Tian, Yu and Liu, Yuyuan and Carneiro, Gustavo},
  journal={arXiv preprint arXiv:2203.01937},
  year={2022}
}