Home

Awesome

NYUTron: Health System Scale Language Models Are All-purpose Prediction Engine

<p align="center"> <a rel="license" href="http://creativecommons.org/licenses/by-nc-sa/2.0/"><img alt="Creative Commons License" style="border-width:0" src="https://i.creativecommons.org/l/by-nc-sa/2.0/88x31.png" /></a> <a href="https://github.com/nyuolab/NYUTron_open/issues"> <img alt="maintenance" src="https://img.shields.io/badge/Maintained%3F-yes-green.svg"> </a> <a href="https://huggingface.co/huggingface"> <img alt="hf" src="https://img.shields.io/badge/Built%20with-%F0%9F%A4%97Huggingface-yellow"> </a> <a href="https://github.com/psf/black"> <img alt="maintenance" src="https://img.shields.io/badge/Code%20style-black-000000.svg"> </a> </p>

This repo contains synthetic data, experimental data and code for the paper "Health System Scale Language Models Are All-purpose Prediction Engine".

See documentation/experiment_and_scripts.md for the scripts corresponding to different experiments in the paper.

Installation: see documentation/installation.md

Deployment: Models were deployed utilizing a modified version of NVIDIA's TRITON Inference Server. The code is available at NYUTriton. Please see manuscript for details.


Paper Authors: Lavender Yao Jiang, Xujin Chris Liu, Nima Pour Nejatian, Mustafa Nasir-Moin, Duo Wang, Anas Abidin, Kevin Eaton, Howard Riina, Ilya Laufer, Paawan Punjabi, Madeline Miceli, Nora C. Kim, Cordelia Orillac, Zane Schnurman, Christopher Livia, Hannah Weiss, David Kurland, Sean Neifert, Yosef Dastagirzada, Douglas Kondziolka, Alexander T.M. Cheung, Grace Yang, Ming Cao, Mona Flores, Anthony B. Costa, Yindalon Aphinyanaphongs, Kyunghyun Cho and Eric Karl Oermann.

Code authors: Lavender Yao Jiang, Eric Karl Oermann, Xujin Chris Liu, Grace Yang, Ming Cao

License: personal/academic use welcome when given credit, contact authors for commercial use. (See LICENSE for details)