tf_unet icon indicating copy to clipboard operation
tf_unet copied to clipboard

Probability of one or both classes tend to get stuck at exactly 50%

Open Snubben-B opened this issue 7 years ago • 4 comments

Hi, I'm trying to classify pixels into one of two categories. I have a problem that the probability of one or both classes tend to get stuck at exactly 50%. It does not happen every time, but very often. I have encountered this problem for several different settings of the hyperparameters and both for adam and momentum. I use cross entropy as the cost function. This occurs even in the very simple case where I have a binary blob as input, and I want the same blob as output. What could cause this? I'm attaching an image of the issue. For this specific case I used layers=5, and features_root=64. image

Snubben-B avatar Oct 30 '17 14:10 Snubben-B

Hard to tell what is causing this. Have you tried to normalize your input?

jakeret avatar Oct 31 '17 19:10 jakeret

I'm not entirely sure of what you mean with normalization in this case. The input is binary, with the values 0 and 1.

Snubben-B avatar Nov 01 '17 09:11 Snubben-B

I was talking about removing the mean and scaling to unit variance.

I'm not sure if I understand your case. You have binary input data and you want to learn a binary classification?

jakeret avatar Nov 02 '17 10:11 jakeret

To clarify, I initially had a more complex case, in which I experienced the same problem. The binary input -> binary output is a simplified case, but the problem persists.

Snubben-B avatar Nov 02 '17 11:11 Snubben-B