Citation

Jahin, Md Abrar and Mridha, M. F. and Dey, Nilanjan and Hossen, Md. Jakir (2026) Human-in-the-Loop Feature Selection Using Interpretable Kolmogorov-Arnold Network-Based Double Deep Q-Network. IEEE Open Journal of the Computer Society, 7. pp. 389-403. ISSN 2644-1268

Text 1.pdf - Published Version Restricted to Repository staff only Download (5MB)

Abstract

Feature selection is critical for improving the performance and interpretability of machine learning models, particularly in high-dimensional spaces where complex feature interactions can reduce accuracy and increase computational demands. Existing approaches often rely on static feature subsets or manual intervention, limiting adaptability and scalability. However, dynamic, per-instance feature selection methods and model-specific interpretability in reinforcement learning remain underexplored. This study proposes a human-in-the-loop (HITL) feature selection framework integrated into a Double Deep Q-Network (DDQN) using a Kolmogorov-Arnold Network (KAN). Our novel approach leverages simulated human feedback and stochastic distribution-based sampling, specifically Beta, to iteratively refine feature subsets per data instance, improving flexibility in feature selection. The KAN-DDQN achieved notable test accuracies of 93% on MNIST and 83% on FashionMNIST, outperforming conventional MLP-DDQN models by up to 9%. The KAN-based model provided high interpretability via symbolic representation while using 4 times fewer neurons in the hidden layer than MLPs did. Comparatively, the models without feature selection achieved test accuracies of only 58% on MNIST and 64% on FashionMNIST, highlighting significant gains with our framework. We further validate scalability on CIFAR-10 and CIFAR-100, achieving up to 30% relative macro F1 improvement on MNIST and 5% on CIFAR-10, while reducing calibration error by 25%. Complexity analysis confirms real-time feasibility with latency below 1 ms and parameter counts under 0.02 M. Pruning and visualization further enhanced model transparency by elucidating decision pathways. These findings present a scalable, interpretable solution for feature selection that is suitable for applications requiring real-time, adaptive decision-making with minimal human oversight.

Item Type:	Article
Uncontrolled Keywords:	Human-in-the-loop (HITL), feature selection, Kolmogorov-Arnold network (KAN), reinforcement learning, double deep Q-network (DDQN)
Subjects:	T Technology > TK Electrical engineering. Electronics Nuclear engineering > TK5101-6720 Telecommunication. Including telegraphy, telephone, radio, radar, television
Divisions:	Faculty of Information Science and Technology (FIST)
Depositing User:	Ms Suzilawati Abu Samah
Date Deposited:	10 Feb 2026 06:29
Last Modified:	10 Feb 2026 06:29
URII:	http://shdl.mmu.edu.my/id/eprint/15307

Downloads

Edit (login required)