Designing an end-to-end machine learning use case

End-to-End Machine Learning

Joshua Stapleton

Machine Learning Engineer

The case study

  • Predicting heart disease
  • Goal: inform decision-making of cardiologists

CardioCare

Doctor with Patient

1 Image source: https://www.flaticon.com/free-icons/doctor
End-to-End Machine Learning

The model's role

  • Models can inform, but should not make decisions
  • Especially important in healthcare

 

A flowchart showing the binary classification process of heart/disease / no heart disease given the patient data

End-to-End Machine Learning

The machine learning lifecycle

  A flowchart showing the machine learning lifecycle: problem understanding, data collection and preparation, model development and tuning, model evaluation, deployment, and finally, monitoring. The process then loops as we refine the sysem

End-to-End Machine Learning

Understanding end user requirements

Accuracy

  A dartboard as a metaphor for accuracy

Reliability

  A chain as a metaphor for reliability

Security

  A magnifying glass as a metaphor for intepretability and looking closer

Interpretability

  A magnifying glass as a metaphor for intepretability and looking closer

End-to-End Machine Learning

Data collection

  • Collect relevant data
    • Private dataset from company
    • Public dataset

 

  • Understand data and context
    • Representation and measurement
    • Potential bias

A graphic showing the data collection process: relevant electronic health records are extracted from either proprietary or public datasets

End-to-End Machine Learning

Let's practice!

End-to-End Machine Learning

Preparing Video For Download...