A Paper Was Accepted by TMI
Ph.D. student Hao Zheng's paper "Alleviating Class-wise Gradient Imbalance for
Pulmonary Airway Segmentation" (by Hao Zheng, Yulei Qin, Yun Gu, Fangfang Xie, Jie
Yang, Jiayuan Sun, Guang-Zhong Yang) was accepted by IEEE Transactions on Medical Imaging, a top journal in the field of medical imaging.
In this paper, the author demonstrates that this problem is arisen by gradient erosion and dilation of the neighborhood voxels. During back-propagation, if the ratio of the foreground gradient to background gradient is small while the class imbalance is local, the foreground gradients can be eroded by their neighborhoods. This process cumulatively increases the noise information included in the gradient flow from top layers to the bottom ones, limiting the learning of small structures in CNNs. To alleviate this problem, the author uses group supervision and the corresponding WingsNet to provide complementary gradient flows to enhance the training of shallow layers. To further address the intra-class imbalance between large and small airways, the author designs a General Union loss function which obviates the impact of airway size by distance-based weights and adaptively tunes the gradient ratio based on the learning process. Extensive experiments on public datasets demonstrate that the proposed method can predict the airway structures with higher accuracy and better morphological completeness than the baselines
（RevisedTime：2021-05-12 14:47 Views：27）
Copyright @2018 Institute of Image Processing and Pattern Recognition, Shanghai Jiao Tong University.
Address: 2-427 SEIEE Building, 800 Dong Chuan Rd, Shanghai, 200240, P.R.China