AI in EE

AI IN DIVISIONS

AI in Signal Division

Thang Vu, Haeyong Kang, Chang D. Yoo, SCNet: Training Inference Sample Consistency for Instance Segmentation, AAAI 2021.

Cascaded architectures have brought significant performance improvement in object detection and instance segmentation. However, there are lingering issues regarding the disparity in the Intersection-over-Union (IoU) distribution of the samples between training and inference. This disparity can potentially exacerbate detection accuracy. This paper proposes an architecture referred to as Sample Consistency Network (SCNet) to ensure that the IoU distribution of the samples at training time is close to that at inference time. Furthermore, SCNet incorporates feature relay and utilizes global contextual information to further reinforce the reciprocal relationships among classifying, detecting, and segmenting sub-tasks. Extensive experiments on the standard COCO dataset reveal the effectiveness of the proposed method over multiple evaluation metrics, including box AP, mask AP, and inference speed. In particular, while running 38% faster, the proposed SCNet improves the AP of the box and mask predictions by respectively 1.3 and 2.3 points compared to the strong Cascade Mask R-CNN baseline.

8

Figure 8. Architecture of cascade approaches: (a) Cascade Mask R-CNN. (b) Hybrid Task Cascade (HTC). (c) the proposed SCNet. Here, “F”, “RPN”, “Pool”, “B”, “M”, “FR” and “G” denote image features, Region Proposal Network, region-wise pooling, box branch, mask branch, feature relay, and global context branch, respectively.