Search papers, labs, and topics across Lattice.
This paper introduces the Single-Stage Hierarchical Rectification (SSHR) framework, which addresses the limitations of traditional multi-stage weakly supervised semantic segmentation methods in histopathology by purifying intermediate feature representations during the forward pass. By implementing a Hierarchical Feature Rectification Module (HFRM), SSHR effectively filters local anomalies using deep global semantic context, resulting in high-fidelity activation maps generated within a single training loop. Experimental results on the LUAD-HistoSeg and BCSS datasets show that SSHR not only outperforms existing state-of-the-art methods but also reduces training time by 2 to 5 times, enhancing computational efficiency for clinical applications.
A single-stage approach to histopathology segmentation cuts training time by up to 5x while achieving superior accuracy compared to traditional multi-stage methods.
Existing weakly supervised semantic segmentation (WSSS) methods in computational pathology rely on a multi-stage paradigm: class activation map (CAM) generation, offline pseudo-mask refinement, and fully supervised retraining. While established, this decoupled approach presents fundamental limitations. The multi-stage process not only incurs high computational training costs but also suffers from error propagation: local texture biases in shallow CNN layers generate false-positive artifacts that subsequent refinement steps often fail to correct. To address these persistent challenges through a simple yet highly effective approach, we propose the Single-Stage Hierarchical Rectification (SSHR) framework. Rather than passively refining CAMs post-hoc, our method proactively purifies intermediate feature representations during the forward pass. We introduce a Hierarchical Feature Rectification Module (HFRM) that utilizes deep global semantic context to filter out local anomalies in shallow layers. This mechanism generates high-fidelity activation maps directly within a single training loop. Experiments on the LUAD-HistoSeg and BCSS datasets demonstrate that SSHR outperforms state-of-the-art multi-stage methods. Furthermore, SSHR reduces training duration by 2 to 5 times. This efficiency minimizes computational overhead and accelerates clinical translation for large-scale histopathology workflows. The code is available at: https://github.com/trongduc-nguyen/SSHR
