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
Explore connections, maps & timelines
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.