Work

Reinforcement Learning for Breast Cancer Diagnosis with Visual Interpretability

Year: 2025

Type: conference-paper

Abstract: This paper presents a reinforcement learning (RL) based approach for breast cancer diagnosis using mammographic images. The diagnosis task is formulated as a binary classification problem between malignant and benign findings. We adopt the Proximal Policy Optimization (PPO) algorithm combined with a custom convolutional neural network (CNN) feature extractor and Grad-CAM-based visual explanations. To emulate a diagnostic workflow, we design a custom multi-step environment where the RL agent processes sequences of mammograms. We report training performance achieving a maximum validation accuracy of 77.1% with mean performance of 72.2% ± 4.5% across five independent runs, classification metrics, and qualitative visualizations of attention maps to demonstrate the model’s interpretability. Our approach bridges the gap between traditional static classifiers and sequential decision-making processes that better reflect clinical diagnostic workflows.

Source: Procedia Computer Science

DOI: 10.1016/j.procs.2025.10.218

Authors: Gani Esen, Marat Nurtas, Laura La Paglia, Aizhan Altaibek, Bazargul Matkerim

Institutions: International Information Technologies University, Al-Farabi Kazakh National University, National Research Council

Language: English

Topic: AI in cancer detection

Subfield: Artificial Intelligence

Field: Computer Science

Domain: Physical Sciences

SDG: Good health and well-being

Open Access status: diamond

Funder: Ministry of Education and Science of the Republic of Kazakhstan