1 min readfrom Machine Learning

3 Collapsing models [R]

Trying to train 3 models for birads detection using cross entropy and center loss + class weights but all of them seem to collapse between birads 1 as the dataset (VinDr) im using is heavily unbalanced towards it, Would like to ask for input and opinion on what seems to be the case, am I using the wrong loss function?

submitted by /u/Rihitwo
[link] [comments]

Want to read more?

Check out the full article on the original site

View original article

Tagged with

#BIRADS detection
#Unbalanced dataset
#Model collapsing
#VinDr
#Cross entropy
#Loss function
#Center loss
#Dataset imbalance
#BIRADS 1
#Class weights
#Data imbalance
#Machine Learning
#Deep Learning
#Training models
#Classification
#Image Classification
#Model optimization
#Model performance
#Radiology
#Medical Imaging