Comment | Published:

How to develop machine learning models for healthcare

Rapid progress in machine learning is enabling opportunities for improved clinical decision support. Importantly, however, developing, validating and implementing machine learning models for healthcare entail some particular considerations to increase the chances of eventually improving patient care.

Access optionsAccess options

Rent or Buy article

Get time limited or full article access on ReadCube.

from$8.99

All prices are NET prices.

References

  1. 1.

    LeCun, Y., Bengio, Y. & Hinton, G. Nature 521, 436–444 (2015).

  2. 2.

    Gulshan, V. et al. JAMA 316, 2402–2410 (2016).

  3. 3.

    Esteva, A. et al. Nature 542, 115–118 (2017).

  4. 4.

    Krause, J. et al. Ophthalmology 125, 1264–1272 (2018).

  5. 5.

    Ehteshami Bejnordi, B. et al. JAMA 318, 2199–2210 (2017).

  6. 6.

    Poplin, R. et al. Nat. Biomed. Eng. 2, 158–164 (2018).

  7. 7.

    Ting, D. S. W. & Wong, T. Y. Nat. Biomed. Eng. 2, 140–141 (2018).

  8. 8.

    Xu, K. et al. Preprint at https://arxiv.org/abs/1502.03044 (2015).

  9. 9.

    Moher, D. et al. BMJ 340, c869 (2010).

  10. 10.

    Japkowicz, N. & Stephen, S. Intell. Data Anal. 6, 429–449 (2002).

  11. 11.

    Rajkomar, A. et al. npj Digit. Med. 1, 18 (2018).

  12. 12.

    Ren, S., He, K., Girshick, R. & Sun, J. IEEE Trans. Pattern Anal. Mach. Intell. 39, 1137–1149 (2017).

  13. 13.

    Liu, Y. et al. Arch. Pathol. Lab. Med. https://doi.org/10.5858/arpa.2018-0147-OA (2018).

  14. 14.

    Steiner, D. F. et al. Am. J. Surg. Pathol. 42, 1636–1646 (2018).

  15. 15.

    De Fauw, J. et al. Nat. Med. 24, 1342–1350 (2018).

  16. 16.

    Sofka, M., Milletari, F., Jia, J. & Rothberg, A. in Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support (eds Cardoso, J. et al.) 258–266 (Springer, 2017).

  17. 17.

    Zoph, B., Vasudevan, V., Shlens, J. & Le, Q. V. Preprint at https://arxiv.org/abs/1707.07012 (2017).

  18. 18.

    Sandler, M., Howard, A., Zhu, M., Zhmoginov, A. & Chen, L.-C. in IEEE Conference on Computer Vision and Pattern Recognition 4510–4520 (IEEE, 2018).

  19. 19.

    Bishop, C. Pattern Recognition and Machine Learning (Springer, 2006).

  20. 20.

    Zhang, C., Bengio, S., Hardt, M., Recht, B. & Vinyals, O. Preprint at https://arxiv.org/abs/1611.03530 (2016).

  21. 21.

    Bergstra, J. & Bengio, Y. J. Mach. Learn. Res. 13, 281–305 (2012).

  22. 22.

    ILSVRC http://www.image-net.org/challenges/LSVRC/announcement-June-2-2015 (2 June 2015).

  23. 23.

    Alba, A. C. et al. JAMA 318, 1377–1384 (2017).

  24. 24.

    Niculescu-Mizil, A. & Caruana, R. in Proc. 22nd International Conference on Machine Learning 625–632 (ACM, 2005).

  25. 25.

    Thabane, L. et al. BMC Med. Res. Methodol. 13, 92 (2013).

  26. 26.

    Parikh, R., Mathai, A., Parikh, S., Chandra Sekhar, G. & Thomas, R. Indian J. Ophthalmol. 56, 45–50 (2008).

  27. 27.

    van Smeden, M., Van Calster, B. & Groenwold, R. H. H. JAMA 319, 1725–1726 (2018).

  28. 28.

    Sayres, R. et al. Ophthalmology 126, 552–564 (2018).

  29. 29.

    Graham, K. C. & Cvach, M. Am. J. Crit. Care 19, 28–34 (2010).

  30. 30.

    Abràmoff, M. D., Lavin, P. T., Birch, M., Shah, N. & Folk, J. C. npj Digit. Med. 1, 39 (2018).

  31. 31.

    Shlens, J. Google AI Blog https://ai.googleblog.com/2016/03/train-your-own-image-classifier-with.html (2016).

Download references

Author information

Correspondence to Po-Hsuan Cameron Chen.

Rights and permissions

Reprints and Permissions

About this article

Verify currency and authenticity via CrossMark
Fig. 1: Examples of different phases of the translational process of developing, validating and implementing ML models for healthcare.
Fig. 2: Dataset naming convention in clinical and ML studies.