Privileged Prior Information Distillation for Image Matting

Authors

  • Cheng Lyu Beijing University of Posts and Telecommunications
  • Jiake Xie PicUP.Ai
  • Bo Xu Xpeng
  • Cheng Lu Xpeng
  • Han Huang AI^2 Robotics
  • Xin Huang Towson University
  • Ming Wu Beijing University of Posts and Telecommunications
  • Chuang Zhang Beijing University of Posts and Telecommunications
  • Yong Tang PicUP.Ai

DOI:

https://doi.org/10.1609/aaai.v38i5.28198

Keywords:

CV: Segmentation, CV: Multi-modal Vision, ML: Applications, ML: Multimodal Learning

Abstract

Performance of trimap-free image matting methods is limited when trying to decouple the deterministic and undetermined regions, especially in the scenes where foregrounds are semantically ambiguous, chromaless, or high transmittance. In this paper, we propose a novel framework named Privileged Prior Information Distillation for Image Matting (PPID-IM) that can effectively transfer privileged prior environment-aware information to improve the performance of trimap-free students in solving hard foregrounds. The prior information of trimap regulates only the teacher model during the training stage, while not being fed into the student network during actual inference. To achieve effective privileged cross-modality (i.e. trimap and RGB) information distillation, we introduce a Cross-Level Semantic Distillation (CLSD) module that reinforces the students with more knowledgeable semantic representations and environment-aware information. We also propose an Attention-Guided Local Distillation module that efficiently transfers privileged local attributes from the trimap-based teacher to trimap-free students for the guidance of local-region optimization. Extensive experiments demonstrate the effectiveness and superiority of our PPID on image matting. The code will be released soon.

Downloads

Published

2024-03-24

How to Cite

Lyu, C., Xie, J., Xu, B., Lu, C., Huang, H., Huang, X., Wu, M., Zhang, C., & Tang, Y. (2024). Privileged Prior Information Distillation for Image Matting. Proceedings of the AAAI Conference on Artificial Intelligence, 38(5), 4044-4052. https://doi.org/10.1609/aaai.v38i5.28198

Issue

Section

AAAI Technical Track on Computer Vision IV