research

Journal Publication

  • Improving Access to Essential Medicines via Decision-Aware Machine Learning
    Chung, A. T.-H., Abdulai, J., Bayoh, P., Sandi, L., Smart, F., Bastani*, H., Bastani*, O. (2026).
    Nature (research article).
    *denote equal last author
    Nature Nature News SSRN Behind the Paper
    News articles
    Abstract

    A critical challenge in healthcare systems in low- and middle-income countries (LMICs) is the efficient and equitable allocation of scarce resources, particularly essential medicines. This problem is complicated by limited high-quality data, which restricts the applicability of traditional data-driven techniques. We propose a novel decision-aware machine learning framework for essential medicines allocation, which additionally leverages multi-task learning to ensure sample efficiency and catalytic priors to ensure equitable allocation. In collaboration with the Sierra Leone national government, we performed a staggered, nationwide deployment of our system as a decision support tool. Our econometric evaluation finds an estimated 19% increase in consumption of allocated products in treated districts, demonstrating its efficacy at improving access to essential medicines. Our tool was subsequently scaled nationwide, covering an estimated 2 million women and children under five. Our work demonstrates how machine learning methods can improve efficiency at very low cost in resource-constrained global health settings.

Working Papers

  • Effective Personalized AI Tutors via LLM-Guided Reinforcement Learning
    Chung, A. T.-H., Zhang, B., Kung, L.-C., Bastani*, H., Bastani*, O.
    Available at SSRN.
    *denote equal last author
    SSRN
    News articles
    Abstract

    Generative AI (GenAI) is rapidly reshaping education by unlocking the potential for personalized tutoring. Yet, emerging platforms largely focus on GenAI chatbot tutors that reactively answer student questions. We hypothesize that the efficacy of GenAI chatbot tutors can be substantially improved by proactively guiding student learning. To test this, we design a novel tutoring platform that tightly integrates a carefully-designed GenAI chatbot with a reinforcement learning algorithm for sequencing practice problems. Critically, this algorithm leverages rich signals from student-chatbot interactions to adaptively select practice problems of an appropriate difficulty level. In partnership with the Taipei City Government and American Institute in Taiwan, we deployed our tutoring platform in conjunction with a five-month course to teach Python to students across ten high schools. We randomized students between a fixed practice problem sequence and our adaptive sequencing algorithm. We find that adaptive sequencing increased unassisted final exam performance by 0.15 standard deviations (equivalent to 6-9 months of schooling by some estimates); mediation analysis suggests that gains were driven by increased engagement. Our work provides large-scale field evidence that student-chatbot interactions provide valuable signals for proactively optimizing and personalizing student learning.

Refereed Conference Papers

  • Decision-Aware Learning for Optimizing Health Supply Chains
    Chung, A. T.-H., Rostami, V., Bastani, H., & Bastani, O. (2022).
    Machine Learning for Health (ML4H).
    arXiv
    Abstract

    We study the problem of allocating limited supply of medical resources in developing countries, in particular, Sierra Leone. We address this problem by combining machine learning (to predict demand) with optimization (to optimize allocations). A key challenge is the need to align the loss function used to train the machine learning model with the decision loss associated with the downstream optimization problem. Traditional solutions have limited flexibility in the model architecture and scale poorly to large datasets. We propose a decision-aware learning algorithm that uses a novel Taylor expansion of the optimal decision loss to derive the machine learning loss. Importantly, our approach only requires a simple re-weighting of the training data, ensuring it is both flexible and scalable, e.g., we incorporate it into a random forest trained using a multitask learning framework. We apply our framework to optimize the distribution of essential medicines in collaboration with policymakers in Sierra Leone; highly uncertain demand and limited budgets currently result in excessive unmet demand. Out-of-sample results demonstrate that our end-to-end approach can significantly reduce unmet demand across 1040 health facilities throughout Sierra Leone.

Book Chapter

  • Optimizing Health Supply Chains in LMICs with Machine Learning: A Case Study in Sierra Leone
    Bastani, H., Bastani, O., & Chung, A. T.-H.. (2024).
    In C. S. Tang (Ed.), Responsible and Sustainable Operations: The New Frontier (pp. 187-202). Springer Nature Switzerland.
    Full Chapter
    Abstract

    This chapter overviews the challenges in pharmaceutical supply chains (PSCs) in Low- and Middle-Income Countries (LMICs), with a focus on Sierra Leone. Furthermore, it describes how traditional supply chain optimization strategies can be used to improve performance of PSCs in Sierra Leone. Finally, it describes the significant potential for using machine learning in this framework for effective demand forecasting. We highlight challenges such as limited data availability, the need to ensure equitable distribution, as well as the potential for transfer learning to address some of these challenges.

Invited Paper

  • Application of AI in Healthcare Management in Developing Countries (in Chinese)
    Chung, A. T.-H.. (2025).
    Development Focus Quarterly, Issue 20.
    Abstract

    Artificial intelligence (AI) is rapidly becoming a key technology for improving healthcare and public health services, especially in developing countries with limited medical resources. This article first integrates international reports and academic literature to summarize the core applications of AI in four areas: disease prevention, telemedicine, healthcare resource allocation, and health education. It then draws on the author’s field experience in Sierra Leone and Somaliland to explain how AI technologies such as decision-aware machine learning, large language models (LLMs), and reinforcement learning (RL) can substantially improve the efficiency and equity of healthcare services.

    Despite its promising prospects, challenges such as insufficient data, weak infrastructure, lack of funding and talent, and incomplete regulatory and ethical frameworks still limit the implementation of AI in resource-constrained settings. To realize the benefits of AI while avoiding risks such as data privacy violations, algorithmic bias, and the widening of healthcare inequality, future development should focus on strengthening infrastructure, cultivating local talent, and establishing appropriate regulatory policies to ensure the ethical and inclusive use of AI, and to achieve the goals of health equity and sustainable development.

Work In Progress

  • Incentive-Compatible Human-AI Collaboration via Adversarial Tasks
    with Bastani, H. and Bastani, O.
  • Operational Outcomes of an AI Medical Scribe: Evidence from Somaliland
    with Qin, J. and Bastani, H.
  • Trust in AI for Resource Allocation Using Housing Images
    with Harari, M. and Wong, M.
  • AI for Poverty Targeting
    with Harari, M. and Wong, M.