Label-efficient Contrastive Learning-based model for nuclei detection and classification in 3D Cardiovascular Immunofluorescent Images
Nazanin Moradinasab, Rebecca A. Deaton, Laura S. Shankman, Gary K. Owens, Donald E. Brown
TL;DR
The paper tackles automated detection and classification of nuclei in 3D multichannel immunofluorescent images under a weakly supervised setting. It introduces LECL, a label-efficient framework that combines Extended Maximum Intensity Projection (EMIP) to generate nucleus-aware 2D representations from 3D data and Supervised Contrastive Learning (SCL) to learn discriminative features, built on the Hover-Net backbone. The training objective blends cross-entropy, Dice, SupCon, and entropy losses as $L = L^{CE}+L^{Dice}+L^{SupCon}+L^{entropy}$ with an explicit $L^{SCL}$ term guiding pixel embeddings, while EMIP uses Voronoi-based boundaries and 3D distance maps to isolate each nucleus in projection. Experiments on D1 and D2 show LECL outperforms MIP-based approaches and prior weakly supervised methods in detection and classification, and ablations confirm the contributions of EMIP and SCL in reducing false overlaps and improving inter-class separability. The work enables more accurate, cost-effective 3D nuclei analysis in cardiovascular pathology and suggests directions for synthetic data, domain adaptation, and application to additional markers.
Abstract
Recently, deep learning-based methods achieved promising performance in nuclei detection and classification applications. However, training deep learning-based methods requires a large amount of pixel-wise annotated data, which is time-consuming and labor-intensive, especially in 3D images. An alternative approach is to adapt weak-annotation methods, such as labeling each nucleus with a point, but this method does not extend from 2D histopathology images (for which it was originally developed) to 3D immunofluorescent images. The reason is that 3D images contain multiple channels (z-axis) for nuclei and different markers separately, which makes training using point annotations difficult. To address this challenge, we propose the Label-efficient Contrastive learning-based (LECL) model to detect and classify various types of nuclei in 3D immunofluorescent images. Previous methods use Maximum Intensity Projection (MIP) to convert immunofluorescent images with multiple slices to 2D images, which can cause signals from different z-stacks to falsely appear associated with each other. To overcome this, we devised an Extended Maximum Intensity Projection (EMIP) approach that addresses issues using MIP. Furthermore, we performed a Supervised Contrastive Learning (SCL) approach for weakly supervised settings. We conducted experiments on cardiovascular datasets and found that our proposed framework is effective and efficient in detecting and classifying various types of nuclei in 3D immunofluorescent images.
