训练菱形参数预测模型损失极小却出现负预测值，求解决及模型建议

数据集与任务背景 #

• 构建了尺寸为200x200的含菱形的合成数据集，数据样例如下：
  
• 任务为预测标签x,y, theta, s_x, s_y，各标签含义：
  • x,y：菱形的中心坐标
  • theta：菱形的旋转角度
  • s_x, s_y：菱形的对角线长度
• 已基于数据集的均值和标准差完成图像归一化，训练2个epoch后的损失曲线如下：
  

当前遇到的问题 #

预测结果中出现了不可能存在的负值，尝试在模型最后一层添加ReLU激活函数后，问题仍未解决。

核心疑问 #

1. 如何调整训练流程或模型，得到能正确预测标签的结果？
2. 当前使用的模型是否存在设计缺陷？
3. 针对该任务，还有哪些可用的替代模型方案？

可复现代码 #

提供了可复现的Google Colab代码（原链接已移除）

内容的提问来源于stack exchange，提问作者Saeed