ST-SACLF: Style Transfer Informed Self-Attention Classifier for Bias-Aware Painting Classification

Mridula Vijendran, Frederick W. B. Li, Jingjing Deng and Hubert P. H. Shum
Communications in Computer and Information Science (CCIS), 2024

ST-SACLF: Style Transfer Informed Self-Attention Classifier for Bias-Aware Painting Classification

Abstract

Painting classification plays a vital role in organizing, finding, and suggesting artwork for digital and classic art galleries. Existing methods struggle with adapting knowledge from the real world to artistic images during training, leading to poor performance when dealing with different datasets. Our innovation lies in addressing these challenges through a two-step process. First, we generate more data using Style Transfer with Adaptive Instance Normalization (AdaIN), bridging the gap between diverse styles. Then, our classifier gains a boost with feature-map adaptive spatial attention modules, improving its understanding of artistic details. Moreover, we tackle the problem of imbalanced class representation by dynamically adjusting augmented samples. Through a dual-stage process involving careful hyperparameter search and model fine-tuning, we achieve an impressive 87.24\% accuracy using the ResNet-50 backbone over 40 training epochs. Our study explores quantitative analyses that compare different pretrained backbones, investigates model optimization through ablation studies, and examines how varying augmentation levels affect model performance. Complementing this, our qualitative experiments offer valuable insights into the model's decision-making process using spatial attention and its ability to differentiate between easy and challenging samples based on confidence ranking.

Downloads

YouTube

Citations

BibTeX

@incollection{vijendran24stsaclf,
 author={Vijendran, Mridula and Li, Frederick W. B. and Deng, Jingjing and Shum, Hubert P. H.},
 booktitle={Communications in Computer and Information Science},
 title={ST-SACLF: Style Transfer Informed Self-Attention Classifier for Bias-Aware Painting Classification},
 year={2024},
 publisher={Springer},
}

RIS

TY  - CHAP
AU  - Vijendran, Mridula
AU  - Li, Frederick W. B.
AU  - Deng, Jingjing
AU  - Shum, Hubert P. H.
T2  - Communications in Computer and Information Science
TI  - ST-SACLF: Style Transfer Informed Self-Attention Classifier for Bias-Aware Painting Classification
PY  - 2024
PB  - Springer
ER  - 

Plain Text

Mridula Vijendran, Frederick W. B. Li, Jingjing Deng and Hubert P. H. Shum, "ST-SACLF: Style Transfer Informed Self-Attention Classifier for Bias-Aware Painting Classification," in Communications in Computer and Information Science, Springer, 2024.

Supporting Grants

Similar Research

Mridula Vijendran, Frederick W. B. Li and Hubert P. H. Shum, "Tackling Data Bias in Painting Classification with Style Transfer", Proceedings of the 2023 International Conference on Computer Vision Theory and Applications (VISAPP), 2023
Daniel Organisciak, Edmond S. L. Ho and Hubert P. H. Shum, "Makeup Style Transfer on Low-Quality Images with Weighted Multi-Scale Attention", Proceedings of the 2020 International Conference on Pattern Recognition (ICPR), 2020
Shuang Chen, Amir Atapour-Abarghouei, Jane Kerby, Edmond S. L. Ho, David C. G. Sainsbury, Sophie Butterworth and Hubert P. H. Shum, "A Feasibility Study on Image Inpainting for Non-Cleft Lip Generation from Patients with Cleft Lip", Proceedings of the 2022 IEEE-EMBS International Conference on Biomedical and Health Informatics (BHI), 2022

 

 

Last updated on 14 April 2024
RSS Feed