Detecting Cancerous Masses in Mammograms
Senior Thesis Project
University of Wisconsin - La Crosse La Crosse, WI
For my senior thesis project I trained CNNs and RNNs to identify
mammogram images that contain cancerous masses using training data provided by the Mayo
Clinic.
- Designed and trained Convolutional Neural Networks (CNNs) and Recurrent Neural Networks (RNNs) using PyTorch Lightning to classify images with high accuracy
- Split the dataset into training, validation, and test sets to ensure accurate model evaluation and prevent overfitting
- Used image preprocessing techniques and transformations to build a robust training set and more accurate model
