У меня есть метки, похожие на глубину, но мне нужно предсказать градиент меток этой глубины. Существуют тысячи значений ~ 0,01 и сотни значений от 1 до 10. Unet отлично подходит для прогнозирования глубины, но не может обучаться на необработанных градиентах.
Я пытался предсказать кумулятивную сумму градиента для обработки несбалансированные значения, но это не помогло. Я также пытался сбалансировать потери с весами: 1-a для малых значений и a для значений, больших 1, где:
a = n_large / n_all
Это также не помогло.
Как я могу точно предсказать градиент?