Skip to content

Display the essential components of machine learning models in a structured way

Notifications You must be signed in to change notification settings

HealthDataInsight/model-card

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

45 Commits
 
 
 
 

Repository files navigation

HDI Template Model Card

What are model cards?

The main purpose of a model card is to display the essential components of models in a structured way.

Model cards are intended to be short documents that contain non-technical and technical content for a variety of audiences. They should be succinct, well-structured and easily digestible by the reader.

Model cards are one approach to increasing transparency between developers, users and stakeholders about details such as any data used, chosen methodology and performance. They are an example of a transparency tool which aims to standardise ethical practice and reporting1.

A model card is not a replacement for detailed documentation covering quality assurance, testing, bias analysis and so on. High-level overview of these areas can be included on a model card where relevant, but any further detail should be accessible from the model card via links to full documentation.

When to use this template

This template can be used at any point during the model development lifecycle. It is recommended to start using and populating a model card as soon as model development begins because it serves as a useful piece of internal documentation.

Once a model is finalised, ready for deployment or put into production, a fully completed model card should be made available alongside the code used for model training.

To achieve full transparency, the code and model card should be published.

How to use this template

Having a template is a helpful starting place for those who may not be familiar with model cards and provides an indication rather than a specification of what information model developers should consider providing. The specific content and structure should be flexed to accommodate specific use cases and models, with the above purpose in mind.

To copy the model card template, you can either:

  1. Download the repository and then copy the 'model_cards' folder into your own repo
  2. Download the model_card_template.md file if you only want the markdown template and not the full folder structure.

You can then start populating the model card with information about the models in use in your system/service/algorithm.

Storing the model card in a repository alongside the code used to train, test and evaluate models will help to keep documentation up to date as both the model and it's documentation can be updated in parallel. If the code is published, then the model card can be made public through the code publication process.

Inspiration

Model cards were first explored in a Google research paper1 in 2018 and are now widely used. This model card format is based on the version used by NHS England.~

The initial design and set-up of this model card template took inspiration from:

The use of model cards was recommended in the UK Government's Generative AI Framework for HMG2, published in January 2024. This specifically references the Google paper1 from which most inspiration was drawn.

Content

This template repository includes:

  1. The model card template
    1. Specification
    2. Intended Use
    3. Data
    4. Methodology and Training
    5. Evaluation and Performance
  2. Visuals folder to store visualisations (such as a confusion matrix) for use in a model card

Footnotes

  1. Model cards for model reporting 2 3

  2. Generative AI Framework for HMG

About

Display the essential components of machine learning models in a structured way

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published