论文标题

开发机器学习算法以诊断与年龄有关的黄斑变性

Developing a Machine-Learning Algorithm to Diagnose Age-Related Macular Degeneration

论文作者

Dua, Ananya, Minh, Pham Hung, Fahmid, Sajid, Gupta, Shikhar, Zheng, Sophia, Moyo, Vanessa, Xue, Yanran Elisa

论文摘要

如今,超过40岁以上的1200万人患有眼部疾病。最常见的是,老年患者容易受到与年龄相关的黄斑变性的影响,这是一种由于视网膜恶化而导致中央视力模糊的眼科疾病。前者只能通过复杂且昂贵的成像软件检测到,这是一个明显的视野测试。这使大量人口患有未经治疗的眼科病,并使它们有完全视力丧失的风险。已经提出了用于治疗眼病的机器学习算法的使用。但是,这些模型的开发受到对适当模型和训练参数的缺乏了解来最大程度地提高模型性能的限制。在我们的研究中,我们通过生成6个模型来解决这些点,每个模型的学习率为1 * 10^n,其中n为0,-1,-2,... -6,并计算了每个模型的F1分数。我们的分析表明,样本不平衡是训练机器学习模型的关键挑战,并且可能导致欺骗性改善培训成本,这并不能转化为模型预测性能的真正改善。考虑到该疾病的广泛影响及其不利影响,我们开发了一种机器学习算法来治疗这种算法。我们在不同的眼病数据集上训练了模型,这些数据集由5000多名患者及其感染眼睛的照片组成。将来,我们希望该模型被广泛使用,尤其是在资源不足的领域,以更好地诊断眼病并改善人类的健康状况。

Today, more than 12 million people over the age of 40 suffer from ocular diseases. Most commonly, older patients are susceptible to age related macular degeneration, an eye disease that causes blurring of the central vision due to the deterioration of the retina. The former can only be detected through complex and expensive imaging software, markedly a visual field test; this leaves a significant population with untreated eye disease and holds them at risk for complete vision loss. The use of machine learning algorithms has been proposed for treating eye disease. However, the development of these models is limited by a lack of understanding regarding appropriate model and training parameters to maximize model performance. In our study, we address these points by generating 6 models, each with a learning rate of 1 * 10^n where n is 0, -1, -2, ... -6, and calculated a f1 score for each of the models. Our analysis shows that sample imbalance is a key challenge in training of machine learning models and can result in deceptive improvements in training cost which does not translate to true improvements in model predictive performance. Considering the wide ranging impact of the disease and its adverse effects, we developed a machine learning algorithm to treat the same. We trained our model on varying eye disease datasets consisting of over 5000 patients, and the pictures of their infected eyes. In the future, we hope this model is used extensively, especially in areas that are under-resourced, to better diagnose eye disease and improve well being for humanity.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源