XAI Personalized Recommendation Algorithm Using ViT and K-Means

被引:1
作者
Cho, Young-Bok [1 ]
机构
[1] Daejeon Univ, Dept Comp & Informat Secur, Daejeon, South Korea
关键词
Unsupervised object segmentation; K-means clustering; Self-supervised learning transformer; Personalized recommendation algorithm; SEGMENTATION;
D O I
10.1007/s42835-024-01843-6
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
This study presents an unsupervised learning instance segmentation methodology using self-supervised learning Vision Transformer (ViT) and K-means methodology. The proposed instance segmentation is attracting attention as a core field of computer vision that assigns all pixels of an image to an appropriate class and localizes objects to bounding boxes. However, the task of producing high-accuracy pixel-level labels is more important than image classification and object detection. It requires high costs and a lot of time. Therefore, this study provides a clear and easy-to-understand explanation for the personalized decision-making process by using an iterative object mask refinement technique that performs class agnostic unsupervised instance segmentation using K-means clustering and self-adaptive supervised learned ViT. do. The proposed method generates pseudo labels that can be used to learn commercial instance segmentation models. The generated pseudo labels have higher accuracy than other methodologies, and the instance segmentation model learned with pseudo labels improves the existing highest performance by more than 50-80%. Therefore, in this study, without changing the structure of the learning function or model, we proposed a personalized ViT recommendation algorithm through single object discovery, multi-object discovery, and supervised learning segmentation using K-means, a simple clustering methodology, and self-supervised learning ViT.
引用
收藏
页码:4495 / 4503
页数:9
相关论文
共 26 条
[1]   An Improved Forest Fire Detection Method Based on the Detectron2 Model and a Deep Learning Approach [J].
Abdusalomov, Akmalbek Bobomirzaevich ;
Islam, Bappy M. D. Siful ;
Nasimov, Rashid ;
Mukhiddinov, Mukhriddin ;
Whangbo, Taeg Keun .
SENSORS, 2023, 23 (03)
[2]   Peeking Inside the Black-Box: A Survey on Explainable Artificial Intelligence (XAI) [J].
Adadi, Amina ;
Berrada, Mohammed .
IEEE ACCESS, 2018, 6 :52138-52160
[3]  
Agarwal A., 2018, AUTOMATED TEST GENER, V1809
[4]   An Unsupervised Learning of Impedance Plethysmograph for Perceiving Cardiac Events (Unsupervised Learning of Impedance Plethysmograph) [J].
Agham, Nishigandha ;
Chaskar, Uttam ;
Samarth, Prachi .
2021 IEEE INTERNATIONAL CONFERENCE ON COMPUTING, COMMUNICATION, AND INTELLIGENT SYSTEMS (ICCCIS), 2021, :470-475
[5]  
Akbari H, 2021, ADV NEUR IN
[6]   Mental Models of Mere Mortals with Explanations of Reinforcement Learning [J].
Anderson, Andrew ;
Dodge, Jonathan ;
Sadarangani, Amrita ;
Juozapaitis, Zoe ;
Newman, Evan ;
Irvine, Jed ;
Chattopadhyay, Souti ;
Olson, Matthew ;
Fern, Alan ;
Burnett, Margaret .
ACM TRANSACTIONS ON INTERACTIVE INTELLIGENT SYSTEMS, 2020, 10 (02)
[7]  
Chen Y-C, 2019, PALAIAHNAKOTE PATTER, V12046, P223
[8]  
cocodataset, MS COCO VALIDATAION
[9]  
Dosovitskiy A., 2021, 9 INT C LEARN REPR I
[10]  
Gianfagna L., 2021, Explainable AI with Python, P81, DOI 10.1007/978-3-030-68640-64