Feb, 2024

基于深度学习的胸部 X 射线严重程度评估的序数回归框架

TL;DR本研究调查了应用顺序回归方法对胸部 X 射线片中的疾病严重程度进行分类。我们提出了一个框架,将顺序回归问题分为三个部分:模型、目标函数和分类函数。我们使用 ResNet50 和 ViT-B-16 深度学习模型应用了不同的编码方法,包括独热编码、高斯编码、进度条编码和我们的软性进度条编码。我们表明编码的选择对性能有很大影响,最佳编码方法取决于所选择的 Cohen's kappa 的加权方式,以及所使用的模型架构。我们在 GitHub 上公开了我们的代码。