My ML Journey Continued
- Sanjay Sundaram
- Oct 10, 2023
- 3 min read
After months of grueling work and endless tweaking, my machine-learning model has finally achieved something beyond my wildest dreams.
So, let’s start from the top. After creating my convolutional neural network (CNN) model, I decided to put it to the test against some of the best professional models out there. These models, typically used by giants like Microsoft and Google, handle massive datasets and are considered top-tier in image classification. The ones I chose to compare mine against were the legendary VGG16 and ResNet50.
First up, VGG16. I stumbled upon an article about how this model skyrocketed to fame around a decade ago. Created by a couple of college students, VGG16 was trained on a million images and competed in a major competition. It categorized images across over a thousand categories with an impressive 96% accuracy. Talk about setting the bar high! This model has been a benchmark in the field, so naturally, I wanted to see how my baby stacked up against it.
Next, there was ResNet50, another CNN but with a twist. ResNet stands for Residual Network, and the "50" denotes the number of layers it has – a deep, complex architecture perfect for large datasets. Microsoft uses ResNet50, which speaks volumes about its robustness. But here's the kicker: deep networks like ResNet50 are often overkill for smaller datasets. This little nugget of information played a big role in my comparison.
With my nerves on edge and excitement bubbling over, I dove into the testing phase. My model was up against the giants, and I was ready to see how it would perform. To my absolute shock, my model held its ground and then some.
Here's the breakdown:
1. VGG16: Despite being a heavyweight in the field, when tested on my smaller dataset, it scored in the high 80s to low 90s. Not bad at all! It showed why it’s still a respected model, consistently delivering solid performance. I was impressed, but my model wasn’t done yet.

2. ResNet50: This one was a surprise. Given its complexity and deep architecture, ResNet50 struggled with my smaller dataset. It scored around 50-60%, barely better than random guessing. It was clear that ResNet50’s strength lies in large datasets, and it was overqualified for my project’s scope.
And then came the moment of truth – my model's performance. My heart was pounding as I ran the tests. To my amazement, my CNN model outperformed ResNet50 and even gave VGG16 a run for its money. The results showed my model achieving accuracy levels comparable to VGG16, sometimes even better. It was a surreal moment, seeing my months of hard work pay off in such a spectacular way.
The realization hit me – my model wasn’t just good; it was exceptional for its intended purpose. It handled the smaller dataset with finesse, extracting meaningful patterns and delivering high accuracy. This was a game-changer. I felt like Tony Stark seeing his Iron Man suit come to life for the first time. Pure exhilaration!
Reflecting on this journey, a few things stood out. First, the importance of understanding the strengths and limitations of different models. VGG16 and ResNet50 are beasts with large datasets, but for smaller, specific tasks, my tailored approach proved more effective. It’s like having the right tool for the job – sometimes a Swiss Army knife beats a chainsaw.
Second, the power of perseverance. There were countless moments of frustration, times when I thought I'd never get the model to work right. But each obstacle was a learning opportunity, each mistake a step closer to success. This journey wasn’t just about coding; it was about growing, adapting, and pushing through challenges.
And lastly, the thrill of discovery. There’s something incredibly satisfying about seeing your creation outperform expectations, especially when you're up against the best. It’s a reminder that innovation often comes from the most unexpected places. Even a 14-year-old tech enthusiast from Dallas can make waves in the world of AI and machine learning.
Looking ahead, I’m buzzing with ideas and possibilities. There’s so much more to explore, so many more projects to tackle. This victory has fueled my passion and confidence. Whether it's refining this model, diving into new applications, or simply dreaming up the next big thing, I’m ready for whatever comes next.
In conclusion, achieving this milestone with my machine learning model has been an unforgettable experience. The future is bright, and I can’t wait to see where this journey takes me next. Stay tuned, because this is just the beginning of an epic adventure!
Comments