Novel Automatic Deep Learning Feature Extractor with Target Class Specific Feature Explanations
File version
Author(s)
Verma, Brijesh
Rahman, Ashfaqur
Wang, Lipo
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
Size
File type(s)
Location
Gold Coast, Australia
License
Abstract
Deep-learning models are popular machine learning models that have gained their popularity in various fields of computer vision, natural language processing etc, due to their excellent predictive accuracy and ability to automatically extract good features from raw input data. Though these models have these significant advantages, most deep learning models are notoriously black-box models. The features learned by these models are not explainable. The need to explain these predictions is important in many high stakes fields. In this work, we propose a method that uses convolutional layers like in Convolutional Neural Networks (CNNs) that automatically learns features from raw data. In our proposed method we extract feature representation that is unique to each target class in the given data. Visualizations of this unique representation explain the feature learned by the model for a specific target class. To classify an input data we use two approaches. The first approach uses the similarity scores to match the features learned for a particular instance with the extracted target class-specific features to make prediction. The second approach uses an intrinsically explainable model like logistic regression to make predictions. Thus our proposed method in addition to having good predictive accuracy, identifies features that are class-specific. The proposed method achieved an accuracy of 99.31 % on MNIST data set, 90.45% on CIFAR-10 data set and 96.45% accuracy on Assira data set.
Journal Title
Conference Title
2023 International Joint Conference on Neural Networks (IJCNN)
Book Title
Edition
Volume
Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject
Deep learning
Artificial intelligence
Data structures and algorithms
Computer Science
Computer Science, Artificial Intelligence
Computer Science, Hardware & Architecture
Engineering
Engineering, Electrical & Electronic
Persistent link to this record
Citation
Kuttichira, DP; Verma, B; Rahman, A; Wang, L, Novel Automatic Deep Learning Feature Extractor with Target Class Specific Feature Explanations, 2023 International Joint Conference on Neural Networks (IJCNN), 2023