Simple Science

Cutting edge science explained simply

# Electrical Engineering and Systems Science# Image and Video Processing# Computer Vision and Pattern Recognition# Machine Learning

Advancements in Breast Cancer Detection Using AI

AI models improve accuracy in predicting breast cancer stages from digital images.

― 4 min read


AI Boosts Breast CancerAI Boosts Breast CancerDetectionpredictions using biopsy images.AI models enhance breast cancer stage
Table of Contents

Cancer is a major health problem around the world. It is the second highest cause of death, following heart disease, with millions of people losing their lives to cancer each year. Breast cancer is one of the most common types of cancer, affecting many women worldwide. By 2020, over 7.8 million women had been diagnosed with breast cancer, making it a serious concern for public health.

Breast Cancer Stages

Breast cancer can be treated effectively if detected early. There are five stages of breast cancer, ranging from Stage 0, which is non-invasive, to Stage 4, which is invasive and has spread to other parts of the body. Mammograms, a type of X-ray of the breast, are commonly used to check for breast cancer. Another method is through a biopsy, where samples of tissue are taken and examined to determine the presence and extent of the disease.

Role of Artificial Intelligence in Breast Cancer Detection

With advances in technology, particularly in Artificial Intelligence (AI) and Deep Learning (DL), researchers are increasingly looking to these tools to aid in the detection of breast cancer. Recent studies have used pre-trained computer vision models to help detect breast cancer from mammograms and other imaging techniques. These models, which include types like ResNet, DenseNet, and U-Net, analyze images of the breast to predict whether cancer is present.

Using Digital Pathology Data

This study uses a dataset known as the Nightingale Open Science dataset, which contains Digital Images of breast biopsy samples. By analyzing these images with pre-trained computer vision models, we aim to predict the stage of breast cancer a patient has. The dataset includes over 72,000 images from thousands of biopsy samples taken from women over several years. This provides a large amount of data to improve the accuracy of Predictions.

Data and Methodology

The images from the biopsies represent different sections of the tissue samples. To predict the overall stage of cancer for a biopsy, we look at the average predictions of each image from that biopsy. We also gather data on the distribution of cancer stages among patients to see how many are diagnosed at each stage.

To prepare the data for the models, we resize the images and split the dataset into parts for training and testing. We fine-tune various pre-trained models to find which one performs best in predicting the cancer stage. Each model is trained with several learning rates to optimize performance.

Results of the Experiments

After training multiple models, we find that EfficientNet outperforms others. This model’s design allows it to effectively balance accuracy and efficiency. We then combine several of the models into what is known as a Deep Ensemble, which generates predictions based on the collective performance of all included models. By doing this, we can achieve even better results in predicting breast cancer stages.

Our main findings indicate that using a deep ensemble approach leads to improved predictions compared to using individual models alone. The ensemble benefits from the strengths of each model and can provide a more reliable prediction for the cancer stage.

Importance of Interpretation and Causality

While achieving accurate predictions is vital, understanding why these models make specific predictions is equally important. Many models struggle when used in real-world situations due to changes in data distribution. To address this, we can apply causal inference methods. These methods help us understand the relationships between various features in the data and how they might influence predictions.

By gaining insights into causal relationships, we can make our models more robust. This will help in reducing biases in predictions and will provide clearer explanations for medical professionals. When doctors understand the factors influencing a model’s prediction, they are more likely to trust the output.

Future Directions

Our research shows promising results in using AI to predict high-risk breast cancer stages. We observe that Deep Ensemble models produce better outcomes than single models. This research also opens the door for further exploration into causal inference methods in medical imaging.

For the future, we aim to investigate how uncertainty estimation techniques can enhance the understanding and interpretation of predictions in medical imaging. By doing so, we hope to improve patient care and outcomes through better detection and understanding of breast cancer.

Conclusion

Breast cancer remains a critical issue in global health, with a significant number of women affected and many lives lost. Utilizing digital pathology images and advanced AI techniques shows great promise in improving cancer detection.

The application of pre-trained computer vision models to analyze biopsy images can assist in predicting cancer stages more accurately. As we continue to refine these models and incorporate methods like causal inference, we can enhance their effectiveness and reliability.

Ultimately, better detection methods may lead to improved treatment and better outcomes for patients diagnosed with breast cancer.

Original Source

Title: Pretrained Vision Models for Predicting High-Risk Breast Cancer Stage

Abstract: Cancer is increasingly a global health issue. Seconding cardiovascular diseases, cancers are the second biggest cause of death in the world with millions of people succumbing to the disease every year. According to the World Health Organization (WHO) report, by the end of 2020, more than 7.8 million women have been diagnosed with breast cancer, making it the world's most prevalent cancer. In this paper, using the Nightingale Open Science dataset of digital pathology (breast biopsy) images, we leverage the capabilities of pre-trained computer vision models for the breast cancer stage prediction task. While individual models achieve decent performances, we find out that the predictions of an ensemble model are more efficient, and offer a winning solution\footnote{https://www.nightingalescience.org/updates/hbc1-results}. We also provide analyses of the results and explore pathways for better interpretability and generalization. Our code is open-source at \url{https://github.com/bonaventuredossou/nightingale_winning_solution}

Authors: Bonaventure F. P. Dossou, Yenoukoume S. K. Gbenou, Miglanche Ghomsi Nono

Last Update: 2023-03-19 00:00:00

Language: English

Source URL: https://arxiv.org/abs/2303.10730

Source PDF: https://arxiv.org/pdf/2303.10730

Licence: https://creativecommons.org/licenses/by/4.0/

Changes: This summary was created with assistance from AI and may have inaccuracies. For accurate information, please refer to the original source documents linked here.

Thank you to arxiv for use of its open access interoperability.

More from authors

Similar Articles