Abstract
Fine-Grained Visual classification (FGVC) is a fundamental problem in computer vision. FGVC is determined by subtle appearance difference of local parts, which thus inspires many part-based methods. Different from attention-based methods and other part-based methods, we propose a novel Category Probability Mask Learning (CPML) module to discover nuanced local differences and mitigate cluttered backgrounds. Meanwhile, the CPML is a simple and efficient module, which can be applied to both convolution neural networks and vision transformers to enhance the ability of feature representation. In addition, we utilize a Category Consistency Loss (CCL) to promote the robustness and discrimination of learned backbone deep features. It is worth mentioning that we only use global branch at test stage because the global feature is already regularized by the part features with CPML and CCL in training steps. Compared with current state-of-the-art methods, our methods achieve promising performance on three widely used FGVC datasets.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Wah, C., Branson, S., Welinder, P., et al.: The Caltech-UCSD birds-200-2011 dataset (2011)
Krause, J., Stark, M., Deng, J., et al.: 3D object representations for fine-grained categorization. In: Proceedings of the IEEE International Conference on Computer Vision Workshops, pp. 554–561 (2013)
Maji, S., Rahtu, E., Kannala, J., et al.: Fine-grained visual classification of aircraft. arXiv preprint arXiv:1306.5151 (2013)
Liu, C., Xie, H., Zha, Z.J., et al.: Filtration and distillation: enhancing region attention for fine-grained visual categorization. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, no. 07, pp. 11555–11562 (2020)
Zheng, H., Fu, J., Mei, T., et al.: Learning multi-attention convolutional neural network for fine-grained image recognition. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 5209–5217 (2017)
Yang, X., Wang, Y., Chen, K., et al.: Fine-grained object classification via self-supervised pose alignment. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7399–7408 (2022)
Wei, X.S., Xie, C.W., Wu, J., et al.: Mask-CNN: localizing parts and selecting descriptors for fine-grained bird species categorization. Pattern Recogn. 76, 704–714 (2018)
Zhang, H., Cao, X., Wang, R.: Audio visual attribute discovery for fine-grained object recognition. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 32, no. 1 (2018)
Xu, H., Qi, G., Li, J., et al.: Fine-grained image classification by visual-semantic embedding. In: IJCAI, pp. 1043–1049 (2018)
Chen, B., Deng, W., Hu, J. Mixed high-order attention network for person re-identification. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 371–381 (2019)
Zhang, Z., Lan, C., Zeng, W., et al.: Relation-aware global attention for person re-identification. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 3186–3195 (2020)
Chen, Y., Dai, X., Liu, M., et al.: Dynamic convolution: attention over convolution kernels. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 11030–11039 (2020)
Hou, Q., Zhou, D., Feng, J.: Coordinate attention for efficient mobile network design. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 13713–13722 (2021)
Qin, Z., Zhang, P., Wu, F., et al.: Fcanet: frequency channel attention networks. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 783–792 (2021)
Wang, Q., Wu, B., Zhu, P., et al.: ECA-Net: efficient channel attention for deep convolutional neural networks. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 11534–11542 (2020)
Dai, J., Qi, H., Xiong, Y., et al.: Deformable convolutional networks. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 764–773 (2017)
Sun, Y., Zheng, L., Yang, Y., et al.: Beyond part models: person retrieval with refined part pooling (and a strong convolutional baseline). In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 480–496 (2018)
Papadopoulos, A., Korus, P., Memon, N.: Hard-attention for scalable image classification. Adv. Neural. Inf. Process. Syst. 34, 14694–14707 (2021)
Woo, S., Park, J., Lee, J.Y., et al.: CBAM: convolutional block attention module. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 3–19 (2018)
Zhang, Y., Xiang, T., Hospedales, T.M., et al.: Deep mutual learning. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4320–4328 (2018)
Xu, S., Chang, D., Xie, J., et al.: Grad-CAM guided channel-spatial attention module for fine-grained visual classification. In: 2021 IEEE 31st International Workshop on Machine Learning for Signal Processing (MLSP), pp. 1–6. IEEE (2021)
Selvaraju, R.R., Cogswell, M., Das, A., et al.: Grad-cam: Visual explanations from deep networks via gradient-based localization. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 618–626 (2017)
Zhuang, P., Wang, Y., Qiao, Y.: Learning attentive pairwise interaction for fine-grained classification. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, no. 07, pp. 13130–13137 (2020)
Wang, Z., Wang, S., Yang, S., et al.: Weakly supervised fine-grained image classification via Guassian mixture model oriented discriminative learning. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 9749–9758 (2020)
Du, R., Chang, D., Bhunia, A.K., et al.: Fine-grained visual classification via progressive multi-granularity training of jigsaw patches. In: Vedaldi, A., Bischof, H., Brox, T., Frahm, J.M. (eds.) Computer Vision–ECCV 2020, pp. 153–168. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-58565-5_10
Dubey, A., Gupta, O., Guo, P., et al.: Pairwise confusion for fine-grained visual classification. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 70–86 (2018)
He, K., Zhang, X., Ren, S., et al.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)
Huang, G., Liu, Z., Van Der Maaten, L., et al.: Densely connected convolutional networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4700–4708 (2017)
Howard, A., Sandler, M., Chu, G., et al.: Searching for mobilenetv3. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 1314–1324 (2019)
Dosovitskiy, A., Beyer, L., Kolesnikov, A., et al.: An image is worth 16x16 words: transformers for image recognition at scale. arXiv preprint arXiv:2010.11929 (2020)
Liu, Z., Lin, Y., Cao, Y., et al.: Swin transformer: hierarchical vision transformer using shifted windows. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 10012–10022 (2021)
Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7132–7141 (2018)
Acknowledgments
This work was supported in part by The National Natural Science Foundation of China (62202061;62171043); Beijing Natural Science Foundation (4232025); R&D Program of Beijing Municipal Education Commission (KM202311232002); Scientific Research Project of National Language Commission (ZDI145-10).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Teng, S., Mei, C., You, X., Lyu, X. (2023). CPML: Category Probability Mask Learning for Fine-Grained Visual Classification. In: Yongtian, W., Lifang, W. (eds) Image and Graphics Technologies and Applications. IGTA 2023. Communications in Computer and Information Science, vol 1910. Springer, Singapore. https://doi.org/10.1007/978-981-99-7549-5_12
Download citation
DOI: https://doi.org/10.1007/978-981-99-7549-5_12
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-7548-8
Online ISBN: 978-981-99-7549-5
eBook Packages: Computer ScienceComputer Science (R0)