At the beginning of this project, I underestimated how challenging medical imaging research would be. While I had some experience with machine learning, working with real-world medical data was very different from anything I had done before. One of the first things I struggled with was simply trying to match the performance of already published models. I initially expected that reproducing similar results would be straightforward, but I quickly realized that even small differences in data handling or setup can lead to noticeable gaps in performance.
Another major challenge I faced was related to the dataset itself. The ChestX-ray14 dataset is highly imbalanced, with far fewer pneumothorax cases compared to normal scans. Early on, I did not fully account for this, and I later discovered issues related to data leakage and improper balancing. This affected my results and made me go back and rethink how I was splitting and evaluating the data. Although frustrating, this was an important learning moment because it showed me how critical proper data handling is in machine learning, especially in medical applications.
I also found it difficult to consistently improve model performance across different training regimes. Reducing the number of positive samples significantly impacted the results, and there were times when long training runs did not lead to meaningful improvements. This was particularly discouraging, but it helped me better understand the limitations of supervised models under low-data conditions.
Working with the self-supervised approach was a turning point. While it required more effort to implement and understand, I found it interesting that it consistently performed better across different settings. This helped me see that improving performance is not just about increasing data, but also about learning better representations from the data itself.
One of the biggest takeaways from this experience is that research is rarely a smooth or linear process. There were multiple points where I had to go back, fix mistakes, and rethink my approach. I also realized the importance of staying organized when running multiple experiments, as it is very easy to lose track of results or repeat the same mistakes.
Overall, this project helped me connect technical concepts to real-world problems. Instead of focusing only on improving metrics, I began to think more about the clinical impact of the work, particularly how better detection of pneumothorax could help reduce missed diagnoses. This experience not only strengthened my technical skills but also gave me a better understanding of how machine learning can be applied in healthcare settings.
