“Building a model that can save as many lives as possible”

How ARA recipient Supreeth Shashikumar is using machine learning to help hospitals detect sepsis — before it’s too late.

Sometimes, good luck wears bad luck's clothing — and that was certainly the case in 2015 for the young electrical and computer engineer Supreeth Shashikumar, when his hunt for a PhD project came up empty. At the Georgia Institute of Technology, no professors were looking for students with his specialty — speech recognition and voice processing.

Supreeth_photo.jpeg
Supreeth Shashikumar, a research scientist at the University of California, San Diego, whose Amazon Research Award supports his research on using machine learning models to predict the onset of sepsis.

So Shashikumar decided to cast his net into a different field and came across a visionary mentor in Shamim Nemati, then an assistant professor in the Department of Biomedical Informatics at the Emory University in Atlanta, where he was also leading the Nemati Lab.

Today, Shashikumar and Nemati are continuing their collaboration at the University of California, San Diego (UCSD), and the medical technology they developed together, supported by an Amazon Research Award, is safeguarding and saving the lives of hospital patients. But let's rewind for a moment.

Back in 2015, when Shashikumar was still stymied in his search, he shifted his focus slightly from speech recognition to the somewhat similar field of biomedical signal processing. The fields overlap in that both are reliant on time series data, such as voice recordings or electrocardiograms.

Shashikumar saw that the Nemati Lab was pioneering the use of time series data held in hospital patients' electronic health records (EHRs) to develop early-warning systems that can aid clinicians by flagging patients who may be on the verge of sudden deterioration. Shashikumar found that to be a compelling idea, so he reached out to Nemati.

“It was a gamble, but it paid off,” Shashikumar says.

Taking on Shashikumar was an easy decision, says Nemati. “Georgia Tech produces some of the best engineers in the country. Add to that somebody who also enjoys bungee jumping and at the same time is extremely detail-oriented, and you’ll get a 10x engineer with a desire to push himself to the limits.” There they started on a multiyear journey toward the development of clinically actionable predictive models in healthcare. Shashikumar would later move with the Nemati Lab to its current home at UCSD.

Related content
Method enforces “coherence” of hierarchical time series, in which the values at each level of the hierarchy are sums of the values at the level below.

The lab’s main focus is the onset of sepsis in hospital emergency departments (EDs), wards, and intensive-care units (ICUs). Sepsis is a sudden and life-threatening condition caused by an infection entering the bloodstream, triggering a catastrophic immune response that can lead to organ failure, septic shock, and death. It is a medical emergency that requires early and aggressive treatment with antibiotics. According to the US-based Sepsis Alliance, for every hour that treatment is delayed, the chance of sepsis moving through severe sepsis to septic shock and death rises by 4%-9%.

ED clinicians are constantly monitoring for signs of sepsis, such as fever and elevated heart rate or respiratory rate. When they suspect sepsis, they order lab tests to look for markers of organ damage. Thus detection, particularly early detection, is crucial.

Shashikumar was drawn to the fact that the Nemati Lab was focused on developing deployable technology. Many researchers take historical time-series patient data from single hospitals and create models to make predictions based on that data, but there is often a chasm between theory and practical deployment, due to the many challenges of working in the healthcare space.

“For us, whenever we pick a project, we are interested in how we can deploy a model into the real world, to do some good by making it clinically actionable,” says Shashikumar. “And, crucially, also make it generalizable.”

This generalizability of medical models is critical if machine learning is to realize its enormous potential benefit to patients.

“Generalizability is about ensuring that your claims about the performance of your model hold in other healthcare system settings,” says Nemati.

For example, say a machine learning model was trained to successfully predict the onset of sepsis in patients in hospital A, using data from that hospital. Could that model then be usefully applied to hospital B with different patient demographics, standards of care, and testing and monitoring procedures? And could it generalize again to hospitals C and D, too? It is an incredibly difficult challenge and one of the reasons for the chasm between research and implementation.

WUPERR

In 2022, in Nature Scientific Reports, Shashikumar and the Nemati team demonstrated that it was indeed possible, with a model called WUPERR (weight uncertainty propagation and episodic representation replay). The model was trained on the EHR data of more than 104,000 patients across four separate healthcare systems. The patient data included over 40 inputs, including ongoing vital signs such as blood pressure and pulse rate, lab test results such as lactate levels in the blood and white blood cell count, patient age, and comorbidities, such as cancer or liver failure.

Related content
ARA recipient Marinka Zitnik is focused on how machine learning can enable accurate diagnoses and the development of new treatments and therapies.

The model overcame several big scientific and practical hurdles inherent to generalization across multiple hospitals: "catastrophic forgetting" and the necessity of keeping patient data confidential.

Catastrophic forgetting is a common problem with transfer learning. When a predictive model is successfully trained on one hospital (hospital A) and then transferred to the next (hospital B), the process will often involve fine-tuning the model on data from hospital B, as no two hospitals are the same. There's a risk, however, that the introduction of new hospital B data will lead the model to "forget" what it learned from hospital A's data.

In theory, one could keep the original model for hospital A and use the fine-tuned model for hospital B, and so on for hospitals C and D. However, not only is this approach impractical, but it also presents a daunting level of regulatory hurdles, according to Shashikumar. Having to deal with a growing number of different models, each of which must meet FDA evaluation and regulation, is simply not scalable.

WUPERR, however, tested a different solution using historical hospital data — a technique called "elastic weight consolidation". This approach echoes a concept found in cognitive neuroscience, according to Shashikumar.

"There are a bunch of neurons in your brain that are trained in the tasks you’ve learned,” he explains. “When you learn a new, similar task, you build on your previous experience — but you don't interfere with those neurons. Instead, you teach additional neurons the nuances of the new task."

With this approach — but with neurons replaced by adjustable model parameters — the team was able to maintain high accuracy in their sepsis predictions across the board with every new hospital added to the pool. By the end, the very accurate sepsis predictions for four hospitals were successfully produced by one model — an important advance.

Related content
Learning the complete quantile function, which maps probabilities to variable values, rather than building separate models for each quantile level, enables better optimization of resource trade-offs.

You may now wonder: How could this model share fiercely protected patient data between separate hospitals? This is the second hurdle that WUPERR overcame, using a technique called “episodic representation replay.” In simple terms, this means that when the model was trained on hospital A's patient data, that data is passed through a neural network that strips away all patient identifiers and creates a representation of the data that is safe to share. The representations of the data are then shown to the model while training at the next hospital.

"I believe this was the first application of sharing neural-network representations from an older hospital with a new hospital in the context of sepsis prediction," says Shashikumar.

The result of all this is a single, manageable model that can generalize across a whole set of hospitals, with all the institutions involved benefiting from each other's patient data while never actually having access to it.

“There is beauty in generalizable knowledge and generalizable models, like a unified theory of everything,” says Nemati.

Things get real

Today the latest iteration of WUPERR is in live action in the ED of a UC San Diego Health hospital, providing clinicians with early warnings about patients predicted to develop sepsis in the next four hours. This version of WUPERR has also been augmented with, among other things, a statistical model that monitors its input data for quality, helping to reduce false alarms.

Related content
With the support of an Amazon Research Award, Papoian’s team is deciphering the dynamics of intrinsically disordered proteins.

That’s important because false alarms are a big problem in sepsis detection. The hospital’s previous, less sophisticated system had a high rate of false alarms. Working with clinicians at the hospital, Shashikumar and his colleagues were able to tune WUPERR to predict 60% of all sepsis events. In the closely monitored environment of the ED, clinicians are expected to catch some portion of the sepsis cases with obvious signs and symptoms, and WUPERR provides a second pair of eyes to provide earlier warning and potentially catch additional cases of sepsis. What is critical to the clinicians is that false alarms, and the burdens they entail, remain low. While about half of WUPERR’s predictions were false alarms, that rate is relatively low, given the seriousness of sepsis.

Missed detections are also of great concern and are often attributable to patient complexity, inadequate monitoring, and low availability of data. Here, the team is applying active sensing to make timely recommendations for collecting sepsis-specific biomarkers in high-risk patients. The latest generation of the system combines false-alarm reduction with active sensing to achieve state-of-the-art performance.

The system has been in place for four months, with data collection ongoing. The clinicians in the ED have reported that, on average, the alarm is going off an hour or two earlier than when the doctors would have started to suspect an infection.

“They’re happy with that performance, particularly the lower false-alarm rate. It’s a very good validation of our work,” says Shashikumar. “But we still have a long way to go. In time, we want to extend this to other hospitals, intensive-care units, and hospital wards across the US and the world.”

The scaling up of this life-saving service is made easier by the fact that WUPERR is entirely cloud-based and hosted on Amazon Web Services.

Related content
New method goes beyond Granger causality to identify only the true causes of a target time series, given some graph constraints.

“Using AWS services has been great for us," says Shashikumar. "Our sepsis software is running in real time in the hospital lab, and that’s mission-critical — it has to be up and running 100% of the time, without fail.” The team makes use of a wide range of AWS services, including autoscaling, load balancing, fault tolerance, and CloudWatch alarms.

Deploying the model in different locations is also greatly simplified. AWS provides HIPAA-compliant infrastructure, which is legally required to protect private health data transmitted to the cloud.

In fact, when the Nemati Lab moved to UC San Diego, they had to decide whether to buy their own in-house servers or move to the cloud. They moved the entirety of their computing services to AWS. “It has been super convenient,” says Shashikumar.

Last year, Nemati's team, including Shashikumar, co-founded Healcisio, a startup, as part of an effort to commercialize their model and ultimately receive FDA clearance, which will be essential for deploying the system to multiple hospitals in the US and abroad.

Meanwhile, they have great ambitions to improve the model. For now, it is limited to the time series data in EHRs. But the team’s current focus is on multimodal data, including wearable sensors, clinical notes, imaging, and more. They want their model to see everything a clinician has access to when they treat patients — all the contextual information — and additionally address “data deserts” via continuous monitoring of patients and active sensing.

Increasing the sensitivity of the model and reducing its false-alarm rate even further is the ultimate goal.

“At the end of the day, our focus is on building a model that can save as many lives as possible,” Shashikumar said. “I didn't get into healthcare out of passion, but it has become my passion.”

Research areas

Related content

US, CA, Palo Alto
We’re working to improve shopping on Amazon using the conversational capabilities of large language models, and are searching for pioneers who are passionate about technology, innovation, and customer experience, and are ready to make a lasting impact on the industry. You’ll be working with talented scientists, engineers, and technical program managers (TPM) to innovate on behalf of our customers. If you’re fired up about being part of a dynamic, driven team, then this is your moment to join us on this exciting journey! We are open to hiring candidates to work out of one of the following locations: Palo Alto, CA, USA
US, WA, Redmond
Project Kuiper is an initiative to launch a constellation of Low Earth Orbit satellites that will provide low-latency, high-speed broadband connectivity to unserved and underserved communities around the world. We are searching for talented candidates with experience in spaceflight trajectory modeling and simulation, orbit mechanics, and launch vehicle mission planning. Key job responsibilities This position requires experience in simulation and analysis of astrodynamics models and spaceflight trajectories. This position requires experience in software development for astrodynamics applications and expertise in supporting mission workflow for satellite operations. Strong analysis skills are required to develop engineering studies of complex large-scale dynamical systems. This position requires demonstrated expertise in computational analysis automation and tool development. Working with the Kuiper engineering team, you will: - Develop modeling techniques for analysis and simulation of deployment dynamics of multiple satellites - Support Project Kuiper’s Launch Vehicle Mission Management team with technical expertise in Launch Vehicle trajectory requirements specification - Develop tools to support Mission Management planning for over 80 launches! - Work collaboratively with launch vehicle system technical teams - Provide support of algorithm development and testing for the Kuiper Flight Dynamics System. - Provide software development support of production code. Export Control Requirement: Due to applicable export control laws and regulations, candidates must be a U.S. citizen or national, U.S. permanent resident (i.e., current Green Card holder), or lawfully admitted into the U.S. as a refugee or granted asylum. We are open to hiring candidates to work out of one of the following locations: Redmond, WA, USA
CN, 11, Beijing
Are you interested in applying your strong quantitative analysis and big data skills to world-changing problems? Are you interested in driving the development of methods, models and systems for strategy planning, transportation and fulfillment network? Are you interested to cooperate with Amazonians around the world? If so, then this is the job for you. Our team, ATE(Analytics Technology and Engineering) is looking for an Applied Scientist to join our growing Science Team in Bangalore (India)/ Beijing(China). We are responsible for creating core analytics tech capabilities, quantative models, platforms development, and data engineering. We develop scalable analytics applications and research models to optimize operations processes. We standardize and optimize data sources and visualization efforts across geographies, build up, and maintain the online business intelligence services and data mart. You will work with other scientists, professional data engineers, business intelligence engineers, and product managers using rigorous quantitative approaches to ensure high quality data tech products for our customers around the world, including India, Australia, Brazil, Mexico, Singapore and Middle East. Amazon is growing rapidly and because we are driven by faster delivery to customers, a more efficient supply chain network, and lower cost of operations, our main focus is in the development of strategic models and automation tools fed by our massive amounts of available data. You will be responsible for building these models/tools that improve the economics of Amazon’s worldwide fulfillment networks in different countries as Amazon increases the speed and decreases the cost to deliver products to customers. You will work on large-scale vehicle routing and scheduling problems under complex operational and physical constraints. You will also identify and evaluate opportunities to reduce variable costs by improving fulfillment center processes, transportation operations and scheduling, and the execution of operational plans. Finally, you will help create the metrics to quantify improvements to the fulfillment costs (e.g., transportation and labor costs) resulting from the application of these optimization models and tools. Key job responsibilities - Design and develop complex mathematical, simulation and optimization models and apply them to define strategic and tactical needs and drive the appropriate business and technical solutions in the areas of vehicle routing, inventory management, network flow, supply chain optimization, demand planning. - Apply theories of mathematical optimization, including linear programming, combinatorial optimization, integer programming, dynamic programming, network flows and algorithms to design optimal or near optimal solution methodologies to be used by in-house decision support tools and software. - Translating business questions and concerns into specific analytical questions that can be answered with available data using Statistical and Machine Learning methods. - Prototype models by using modeling and programming languages with efficient data querying and modeling infrastructure. - Communicate proposals and results in a clear manner backed by data and coupled with actionable conclusions to drive business decisions. - Collaborate with colleagues from multidisciplinary science, engineering and business backgrounds. - Manage your own process. Prioritize and execute on high impact projects, triage external requests, and ensure to deliver projects in time. We are open to hiring candidates to work out of one of the following locations: Beijing, 11, CHN
GB, Cambridge
Our team builds generative AI solutions that will produce some of the future’s most influential voices in media and art. We develop cutting-edge technologies with Amazon Studios, the provider of original content for Prime Video, with Amazon Game Studios and Alexa, the ground-breaking service that powers the audio for Echo. Do you want to be part of the team developing the future technology that impacts the customer experience of ground-breaking products? Then come join us and make history. We are looking for a passionate, talented, and inventive Applied Scientist with a background in Machine Learning to help build industry-leading Speech, Language, Audio and Video technology. As an Applied Scientist at Amazon you will work with talented peers to develop novel algorithms and generative AI models to drive the state of the art in audio (and vocal arts) generation. Position Responsibilities: * Participate in the design, development, evaluation, deployment and updating of data-driven models for digital vocal arts applications. * Participate in research activities including the application and evaluation and digital vocal and video arts techniques for novel applications. * Research and implement novel ML and statistical approaches to add value to the business. * Mentor junior engineers and scientists. We are open to hiring candidates to work out of one of the following locations: Cambridge, GBR | London, GBR
GB, Cambridge
Our team undertakes research together with multiple organizations to advance the state-of-the-art in speech technologies. We not only work on giving Alexa, the ground-breaking service that powers Echo, her voice, but we also develop cutting-edge technologies with Amazon Studios, the provider of original content for Prime Video. Do you want to be part of the team developing the latest technology that impacts the customer experience of ground-breaking products? Then come join us and make history. We are looking for a passionate, talented, and inventive Senior Applied Scientist with a background in Machine Learning to help build industry-leading Speech, Language and Video technology. As a Senior Applied Scientist at Amazon you will work with talented peers to develop novel algorithms and modelling techniques to drive the state of the art in speech and vocal arts synthesis. Position Responsibilities: * Participate in the design, development, evaluation, deployment and updating of data-driven models for digital vocal arts applications. * Participate in research activities including the application and evaluation and digital vocal and video arts techniques for novel applications. * Research and implement novel ML and statistical approaches to add value to the business. * Mentor junior engineers and scientists. We are open to hiring candidates to work out of one of the following locations: Cambridge, GBR | London, GBR
US, VA, Arlington
The GenAI Innovation Center helps AWS customers accelerate their use of Generative AI to solve business challenges and promote innovation across their organizations. The Public Sector team focuses on public sector customers and their unique challenges. As a data scientist, you have deep and broad experience as an ML practitioner. You interface directly with customers to understand and identify their challenges that can be addressed by Generative AI. You build secure solutions that can scale to the size of the problem at hand and guide customers through your rigorous evaluation process. You'll design and run experiments, research new algorithms, and find new ways of optimizing risk, profitability, and customer experience. You're part of both a small team dedicated to public sector customers and a global organization enabling customers to accelerate their progress on GenAI. This position requires that the candidate selected be a US Citizen. AWS Sales, Marketing, and Global Services (SMGS) is responsible for driving revenue, adoption, and growth from the largest and fastest growing small- and mid-market accounts to enterprise-level customers including public sector. The AWS Global Support team interacts with leading companies and believes that world-class support is critical to customer success. AWS Support also partners with a global list of customers that are building mission-critical applications on top of AWS services. Key job responsibilities The primary responsibilities of this role are to: Design, develop, and evaluate innovative ML models to solve diverse challenges and opportunities across industries. Interact with customer directly to understand their business problems, and help them with defining and implementing scalable Generative AI solutions to solve them. Work closely with account teams, research scientist teams, and product engineering teams to drive model implementations and new solution. A day in the life 1. Team with a GenAI strategist to understand a customer problem and provide guidance on how and whether GenAI can help address the issue. 2. Share your latest experiment results or challenges with other scientists on the team. 3. Collaborate on a blog post to share the results and methods used in your most recent customer success. 4. Attend or a deliver a tech talk to highlight a project you or a team mate just completed. 5. Provide feedback to your team during a code review. 6. Meet with customer stakeholders to demonstrate the latest progress on their problem. About the team Diverse Experiences AWS values diverse experiences. Even if you do not meet all of the qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying. Why AWS? Amazon Web Services (AWS) is the world’s most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating — that’s why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses. Inclusive Team Culture Here at AWS, it’s in our nature to learn and be curious. Our employee-led affinity groups foster a culture of inclusion that empower us to be proud of our differences. Ongoing events and learning experiences, including our Conversations on Race and Ethnicity (CORE) and AmazeCon (gender diversity) conferences, inspire us to never stop embracing our uniqueness. Mentorship & Career Growth We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional. Work/Life Balance We value work-life harmony. Achieving success at work should never come at the expense of sacrifices at home, which is why flexible work hours and arrangements are part of our culture. When we feel supported in the workplace and at home, there’s nothing we can’t achieve in the cloud. We are open to hiring candidates to work out of one of the following locations: Arlington, VA, USA | Denver, CO, USA
US, CA, Santa Clara
Amazon is looking for a motivated individual with strong analytical and algorithmic skills and practical experience to join the Modeling and Optimization (MOP) Routing Science team. Your main focus will be on developing and improving our last-mile experience, with emphasis on algorithmic and analytical work. We are looking for candidates with proven ability to design, implement, and evaluate state-of-the-art solutions to large-scale optimization problems, working closely with software development engineers. The position requires strong background in combinatorial optimization, algorithms, algorithm engineering, and data structures, particularly as it applies to vehicle routing and related problems. Familiarity with data science and Machine Learning techniques is a plus. You will also play an integral role in the network planning, modeling, and analysis that will improve the efficiency and cost effectiveness of global fulfillment operations. You will identify and evaluate opportunities to reduce variable costs by improving the transportation network topology, inventory placement, transportation operations and scheduling, fulfillment center processes, and the execution to operational plans. You will also improve the efficiency of capital investment by helping plan the location and deployment of fixed assets. Finally, you will help create the metrics to quantify improvements to the fulfillment costs (e.g., transportation and labor costs) resulting from the application of these optimization models and tools. Key job responsibilities We are looking for candidates with proven ability to design, implement, and evaluate state-of-the-art solutions to large-scale optimization problems, working closely with software development engineers. The position requires strong background in combinatorial optimization, algorithms, algorithm engineering, and data structures, particularly as it applies to vehicle routing problems. We are open to hiring candidates to work out of one of the following locations: Santa Clara, CA, USA
US, WA, Seattle
We are looking for detail-oriented, organized, and responsible individuals who are eager to learn how to work with large and complicated data sets. Some knowledge of econometrics, as well as basic familiarity with Python is necessary, and experience with SQL and UNIX would be a plus. These are full-time positions at 40 hours per week, with compensation being awarded on an hourly basis. You will learn how to build data sets and perform applied econometric analysis at Internet speed collaborating with economists, scientists, and product managers. These skills will translate well into writing applied chapters in your dissertation and provide you with work experience that may help you with placement. Roughly 85% of previous cohorts have converted to full time economics employment at Amazon. If you are interested, please send your CV to our mailing list at econ-internship@amazon.com. We are open to hiring candidates to work out of one of the following locations: Seattle, WA, USA
US, WA, Seattle
We are looking for detail-oriented, organized, and responsible individuals who are eager to learn how to work with large and complicated data sets. Some knowledge of econometrics, as well as basic familiarity with Python is necessary, and experience with SQL and UNIX would be a plus. These are full-time positions at 40 hours per week, with compensation being awarded on an hourly basis. You will learn how to build data sets and perform applied econometric analysis at Internet speed collaborating with economists, scientists, and product managers. These skills will translate well into writing applied chapters in your dissertation and provide you with work experience that may help you with placement. Roughly 85% of previous cohorts have converted to full time economics employment at Amazon. If you are interested, please send your CV to our mailing list at econ-internship@amazon.com. We are open to hiring candidates to work out of one of the following locations: Seattle, WA, USA
US, MA, North Reading
We are looking for experienced scientists and engineers to explore new ideas, invent new approaches, and develop new solutions in the areas of Controls, Dynamic modeling and System identification. Are you inspired by invention? Is problem solving through teamwork in your DNA? Do you like the idea of seeing how your work impacts the bigger picture? Answer yes to any of these and you’ll fit right in here at Amazon Robotics. We are a smart team of doers that work passionately to apply cutting edge advances in robotics and software to solve real-world challenges that will transform our customers’ experiences in ways we can’t even imagine yet. We invent new improvements every day. We are Amazon Robotics and we will give you the tools and support you need to invent with us in ways that are rewarding, fulfilling and fun. Key job responsibilities Applied Scientists take on big unanswered questions and guide development team to state-of-the-art solutions. We want to hear from you if you have deep industry experience in the Mechatronics domain and : * the ability to think big and conceive of new ideas and novel solutions; * the insight to correctly identify those worth exploring; * the hands-on skills to quickly develop proofs-of-concept; * the rigor to conduct careful experimental evaluations; * the discipline to fast-fail when data refutes theory; * and the fortitude to continue exploring until your solution is found We are open to hiring candidates to work out of one of the following locations: North Reading, MA, USA | Westborough, MA, USA