Maintenance Notice

Due to necessary scheduled maintenance, the JMIR Publications website will be unavailable from Wednesday, July 01, 2020 at 8:00 PM to 10:00 PM EST. We apologize in advance for any inconvenience this may cause you.

Who will be affected?

Accepted for/Published in: JMIR Formative Research

Date Submitted: Jul 28, 2022
Date Accepted: Jan 31, 2023

The final, peer-reviewed published version of this preprint can be found here:

Calibrating a Transformer-Based Model’s Confidence on Community-Engaged Research Studies: Decision Support Evaluation Study

Ferrell B, Raskin SE, Zimmerman EB

Calibrating a Transformer-Based Model’s Confidence on Community-Engaged Research Studies: Decision Support Evaluation Study

JMIR Form Res 2023;7:e41516

DOI: 10.2196/41516

PMID: 36939830

PMCID: 10131979

Calibrating Transformer-Based Model’s Confidence on Community-Engaged Research Studies: Decision Support Evaluation Study

  • Brian Ferrell; 
  • Sarah E. Raskin; 
  • Emily B. Zimmerman

ABSTRACT

Background:

Deep learning offers great benefits in classification tasks such as medical imaging diagnostics or stock trading, especially when compared to human-level performances and can be a viable option for classifying distinct levels within Community-engaged research (CEnR). CEnR is a collaborative approach between academics and community partners with the aim of conducting research that is relevant to community needs while incorporating diverse forms of expertise. In the field of deep learning and AI, training multiple models to get the highest validation accuracy is common practice; however, it can overfit towards that specific dataset and not generalize well to a real-world population which creates issues of bias and potentially dangerous algorithmic decisions. Because of this there is a need for creating techniques and exhaustive evaluative processes for these powerful unexplainable models if we plan on automating human decision-making so that we can ensure we do not incorporate and blindly trust poor AI models with real world decisions.

Objective:

We propose an evaluation study to see if our most accurate transformer-based models derived from previous studies can emulate our own classification spectrum for tracking community-engaged research studies as well as if the use of calibrated confidence scores are meaningful.

Methods:

We compare the results from three domain experts, classifying a sample of 45 studies derived from our university’s IRB database to three previously trained transformer-based models, as well as showing if calibrated confidence scores can be a viable technique for using AI as a support role for complex decision-making systems.

Results:

Our findings show that certain models display greater confidence despite not having the highest validation accuracy.

Conclusions:

There is a need to further explore methods that allow domain experts to “trust” our model more.


 Citation

Please cite as:

Ferrell B, Raskin SE, Zimmerman EB

Calibrating a Transformer-Based Model’s Confidence on Community-Engaged Research Studies: Decision Support Evaluation Study

JMIR Form Res 2023;7:e41516

DOI: 10.2196/41516

PMID: 36939830

PMCID: 10131979

Download PDF


Request queued. Please wait while the file is being generated. It may take some time.

© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.