[Repost Q] Request feedback on my unet training: is this underfitting or overfitting?

Not OP. This question is being reposted to preserve technical content removed from elsewhere. Feel free to add your own answers/discussion.

Original question:

The more I read, the more I am confused as to how to interpret the validation and training loss graphs, so therefore I would like to ask for some guidance on how to interpret these values here in the picture. I am training a basic UNet architecture. I am now wondering if I need a more complex network model, or that I just need more data to improve the accuracy.

Historical note: I had the issue where validation loss was exploding after a few epochs, but I added dropout layers and that seems to have fixed the situation.

My current interpretation is that the validation loss is slowly increasing, so does that mean that it’s useless to train further? Or should I rather let it train further because the validation accuracy seems to sometimes jump up a little bit?

Welcome! This is a place for people to learn more about machine learning techniques, discuss applications and ask questions.

Example questions:

“Should I use a deep neural network for my audio classification task?”
“I’m working with a small dataset, what can I do to make my model generalize well?”
“Is there a library available that implements function X in language Y?”
“I want to learn more about the math behind machine learning technique A, where should I start?”

Please do:

Be kind to new people
Post guides and tutorials that you find helpful
Link to open/free sources instead of paywalled when possible

Please don’t:

Post news articles / memes (there are other machine learning/AI communities for this)

Other communities in this area:

Similar subreddits: r/MLquestions, r/askmachinelearning, r/learnmachinelearning

[Repost Q] Request feedback on my unet training: is this underfitting or overfitting?(i.stack.imgur.com)

Learn Machine Learning

!learnmachinelearning@sh.itjust.works

Community stats

Community moderators