DEGPR: Deep Guided Posterior Regularization for Multi-Class Cell Detection and Counting

Abstract

Multi-class cell detection and counting is an essential task for many pathological diagnoses. Manual counting is tedious and often leads to inter-observer variations among pathologists. While there exist multiple, general-purpose, deep learning-based object detection and counting methods, they may not readily transfer to detecting and counting cells in medical images, due to the limited data, presence of tiny overlapping objects, multiple cell types, severe class-imbalance, minute differences in size/shape of cells, etc. In response, we propose guided posterior regularization DeGPR, which assists an object detector by guiding it to exploit discriminative features among cells. The features may be pathologist-provided or inferred directly from visual data. We validate our model on two publicly available datasets (CoNSeP and MoNuSAC), and on MuCeD, a novel dataset that we contribute. MuCeD consists of 55 biopsy images of the human duodenum for predicting celiac disease. We perform extensive experimentation with three object detection baselines on three datasets to show that DeGPR is model-agnostic, and consistently improves baselines obtaining up to 9% (absolute) mAP gains.

Model Details

Yolov5 is trained on MuCeD dataset with Posterior Regularization.

Model Sources [optional]

Repository: [https://github.com/dair-iitd/DeGPR]
Paper : here

Training Details

Training Data

Training dataset can be found here

Results

Model	Precision	Recall	mAP	MAE IEL	MRE IEL	MAE Epith	MRE Epith
Yolov5	0.711	0.723	0.751	8.97	42.62	14.61	13.43
Yolov5(DeGPR)	0.744	0.735	0.787	5.83	24.19	13.15	12.46
Faster-RCNN	0.592	0.436	0.496	11.85	50.05	27.50	24.93
Faster-RCNN (DeGPR)	0.646	0.468	0.541	9.61	31.64	26.50	23.60
EfficientDet	0.266	0.640	0.414	20.35	133.91	20.30	20.78
EfficientDet (DeGPR)	0.274	0.641	0.425	17.32	90.04	18.51	18.12

Citation

@inproceedings{tyagi2023degpr,
  title={DeGPR: Deep Guided Posterior Regularization for Multi-Class Cell Detection and Counting},
  author={Tyagi, Aayush Kumar and Mohapatra, Chirag and Das, Prasenjit and Makharia, Govind and Mehra, Lalita and AP, Prathosh and others},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={23913--23923},
  year={2023}
}