Few Shot Learning Geometric Ensemble for Multi-label Chest X-Rays

Authors: Dana Moukheiber⋆, Saurabh Mahindre⋆, Lama Moukheiber, Mira Moukheiber, Song Wang, Chunwei Ma, George Shih, Yifan Peng, and Mingchen Gao

The dataset labels are available in the labels folder. at: Google Drive Link

The weights for pretrained base feature extractor (ResNet) model are available here: Google Drive Link

• Deepvoro.ipynb: Jupyter notebook with code for:

  • Loading model weights
  • DC Voronoi LR
  • NCA Loss finetuning
  • BCE Loss finetuning
  • Episodic evaluation
  • Finetuning
  • Deepvoro ensemble
  • Few-shot evaluation

• few_shot_cxray: Module with code for:

  • Residual Network backbone
  • Multilabel NCA Loss definition
  • Dataset loaders
  • Utilities