Foreground-aware Virtual Staining for Accurate 3D Cell Morphological Profiling
This work improves virtual staining for 3D cell morphological profiling, which is incremental as it enhances existing methods for better downstream tasks like segmentation.
The paper tackled the problem of virtual staining for 3D cell imaging by addressing the issue of existing methods reproducing background noise instead of focusing on cellular structures, resulting in improved morphological representation and pixel-level accuracy on a benchmark dataset.
Microscopy enables direct observation of cellular morphology in 3D, with transmitted-light methods offering low-cost, minimally invasive imaging and fluorescence microscopy providing specificity and contrast. Virtual staining combines these strengths by using machine learning to predict fluorescence images from label-free inputs. However, training of existing methods typically relies on loss functions that treat all pixels equally, thus reproducing background noise and artifacts instead of focusing on biologically meaningful signals. We introduce Spotlight, a simple yet powerful virtual staining approach that guides the model to focus on relevant cellular structures. Spotlight uses histogram-based foreground estimation to mask pixel-wise loss and to calculate a Dice loss on soft-thresholded predictions for shape-aware learning. Applied to a 3D benchmark dataset, Spotlight improves morphological representation while preserving pixel-level accuracy, resulting in virtual stains better suited for downstream tasks such as segmentation and profiling.