arXiv · 2406.01821
GPU-Accelerated Rule Evaluation and Evolution
Abstract
This paper introduces an innovative approach to boost the efficiency and scalability of Evolutionary Rule-based machine Learning (ERL), a key technique in explainable AI. While traditional ERL systems can distribute processes across multiple CPUs, fitness evaluation of candidate rules is a bottleneck, especially with large datasets. The method proposed in this paper, AERL (Accelerated ERL) solves this problem in two ways. First, by adopting GPU-optimized rule sets through a tensorized representation within the PyTorch framework, AERL mitigates the bottleneck and accelerates fitness evaluation significantly. Second, AERL takes further advantage of the GPUs by fine-tuning the rule coefficients via back-propagation, thereby improving search space exploration. Experimental evidence confirms that AERL search is faster and more effective, thus empowering explainable artificial intelligence.
Explore related subjects
Keep this discovery
Hormoz Shahrzad, Risto Miikkulainen. 2024-06-03. GPU-Accelerated Rule Evaluation and Evolution. https://doi.org/10.1145/3712255.3736397
Cite the original work for its findings. Save a collection to share your selection of sources.