Senior Applied Scientist · AWS GenAI Innovation Center

Jae Oh Woo

Reliable Reasoning, Calibration, and Uncertainty for AI Systems

I build the foundations and systems needed to decide when AI reasoning can be trusted, when it should defer, and how uncertainty turns into safer decisions — combining calibration, self-evaluation, and distribution-free guarantees for real LLM and agentic systems.

Selected Work CV (Upon Request) Google Scholar
Jae Oh Woo

Senior Applied Scientist

GenAI Innovation Center, AWS

Ph.D. Applied Mathematics, Yale

Research Recognition

Industry Impact

Chalk Talk · 500-level AWS re:Invent · INV508 LLMs Reflecting on Their Reasoning
Chalk Talk · 500-level AWS re:Invent · INV506 Bayesian Multi-Path Uncertainty in LLMs
Builder Workshop AWS re:Invent · SPS301 Robo Reviewer for AI Video Evaluation

Service & Awards

Gold Reviewer ICML 2026 Reviewer Award
Top Reviewer NeurIPS 2023 Reviewer Recognition
Session Chair · Reviewer ICLR · ICML · NeurIPS · ISIT Program Committee Service

2026

2025

  • Accepted LLM self-evaluation via VC framework at NeurIPS 2025 Workshop.
  • Talk INV506 — Known Unknowns: Bayesian Multi-Path Framework for Uncertainty in LLMs at AWS re:Invent with Baishali Chaudhury.
  • Talk INV508 — LLMs Reflecting on Their Reasoning: A Probabilistic VC-Theory Approach at AWS re:Invent with Mengdie Flora Wang.
  • Workshop SPS301 — Robo Reviewer: Building AI Video Evaluators with Amazon Bedrock at AWS re:Invent with Haochen Xie, Mengdie Flora Wang, Baishali Chaudhury, Chun-Hao Liu.

2024

  • Accepted Instruction following via uncertainty estimation at ICML 2024.

2023

  • Award Samsung Best Paper Award, Gold Prize.
  • Service Selected as NeurIPS 2023 Top Reviewer.

Modern AI systems should not only answer; they should know when their reasoning is reliable, when uncertainty is high, and when decisions should be deferred.

I study how AI systems reason, quantify uncertainty, and improve under limited supervision. At AWS, I develop methods that make large language models more reliable, better calibrated, and grounded in principled foundations.

My research combines mathematical rigor with practical AI systems — spanning LLM self-evaluation, Bayesian uncertainty, active learning, and information-theoretic foundations.

Before AWS, I worked at Samsung SDS Research America and Moody’s Analytics. I hold a Ph.D. in Applied Mathematics from Yale, a B.S. in Mathematics from KAIST, and was a Simons Postdoctoral Fellow at UT Austin.

My goal is to bridge mathematical rigor and real-world AI systems in ways that make advanced models more trustworthy and useful.

LLM Reasoning & Self-Evaluation

Measuring when LLMs can evaluate their own reasoning — logical consistency, self-correction, and verification.

self-correctionlogical consistencyverification
Featured: Can LLMs Reliably Evaluate Themselves?

Uncertainty & Calibration

Calibration, confidence, and reliability under limited supervision — Bayesian and information-theoretic approaches.

uncertaintyactive learningreliability
Featured: Balanced Entropy Active Learning

Mathematical Foundations

Information theory, entropy, Bayesian learning, and VC-style analysis — rigorous foundations for learning and inference.

information theoryentropylearning theory
Featured: Analytic Mutual Information in BNNs
ICML 2026 Workshop

From Debate to Decision: Distribution-Free Act-or-Defer Control for Multi-Agent LLM Pipelines

Problem Multi-agent debate can converge to confident but wrong consensus — agreement is not correctness.
Method Conformal social choice turns debate outputs into calibrated act-or-defer decisions with distribution-free risk control.
Impact Converts uncertain multi-agent deliberation into safe deployment decisions that know when to escalate.
MF Wang, H Xie, G Wang, A Gao, G Yang, Z Li, QW Qiu, F Han, H Qiu, ..., JO Woo
EACL 2026 Findings

Reflect, Rewrite, Repeat: How Simple Arithmetic Enables Advanced Reasoning in Small Language Models

Problem Small LMs lack the reasoning depth of large models, yet are critical for cost-efficient deployment.
Method Iterative self-reflection on simple arithmetic bootstraps advanced reasoning patterns.
Impact Closes the reasoning gap for small models without large-scale pretraining — viable LLM reliability at the edge.
MF Wang, H Xie, MY Kim, B Chaudhury, M Ashok, S Gunturu, S Hong, JO Woo, ...
NeurIPS 2025 Workshop

Can LLMs Reliably Evaluate Themselves? A Probabilistic VC Framework

Problem Black-box LLM self-evaluation is widely used, but its generalization limits are not well understood.
Method A probabilistic VC framework bounds the sample complexity and capacity of self-assessment.
Impact Estimates when black-box self-evaluation can generalize beyond observed tasks — exposing a capacity-calibration tension.
JO Woo, MF Wang, R Ghosh, B Chaudhury, MY Kim
ICML 2024

Improving Instruction Following in Language Models through Proxy-Based Uncertainty Estimation

Problem LLMs inconsistently follow instructions, especially under ambiguity.
Method Proxy-based uncertainty estimation to flag low-confidence instruction adherence at inference time.
Impact Adds a calibration layer for instruction following without retraining the base model — deployable to any frozen LLM.
J Lee, JO Woo, J Seok, P Hassanzadeh, W Jang
ICLR 2023 Top 25%

Active Learning in Bayesian Neural Networks with Balanced Entropy Learning Principle

Problem Standard acquisition functions are biased toward extremes of model uncertainty — either too confident or too unsure.
Method A balanced-entropy criterion derived from information-theoretic first principles.
Impact Foundational principle for data-efficient learning under limited supervision — transferred into 4 issued US patents at Samsung.
JO Woo
Presenting Balanced Entropy at ICLR 2023 Samsung SDS Gold Award trophy for Balanced Entropy paper
IEEE ISIT 2022

Analytic Mutual Information in Bayesian Neural Networks

Problem Mutual information in BNNs requires expensive approximation.
Method Derives a closed-form expression via information-theoretic analysis.
Impact Bridges information theory foundations with practical deep learning.
JO Woo

See all publications ↓

Short essays on reliability, calibration, structural uncertainty, and decision-making in AI systems — the worldview behind my research.

Also on Google Scholar.

  1. 2026

    Knowing When to Ask: Self-Gated Clarification for Hierarchical Language Agents

    A Gao, Y Kang, MF Wang, JO Woo · ACM CAIS 2026 Workshop on AI Agents for Discovery in the Wild Workshop arXiv

  2. 2026

    From Debate to Decision: Distribution-Free Act-or-Defer Control for Multi-Agent LLM Pipelines

    MF Wang, H Xie, G Wang, A Gao, G Yang, Z Li, QW Qiu, F Han, H Qiu, ... · ICML 2026 Workshop on Statistical Frameworks for Uncertainty in Agentic Systems Workshop arXiv

  3. 2026

    Quantifying Consistency in LLM Logical Reasoning via Structural Uncertainty

    B Chaudhury, MF Wang, HH Park, R Ghosh, S Hong, JO Woo · ICLR 2026 Workshop paper

  4. 2026

    Reflect, Rewrite, Repeat: How Simple Arithmetic Enables Advanced Reasoning in Small Language Models

    MF Wang, H Xie, MY Kim, B Chaudhury, M Ashok, S Gunturu, S Hong, ... · EACL 2026 Findings Paper Code

  5. 2025

    Can LLMs Reliably Evaluate Themselves? A Probabilistic VC Framework

    JO Woo, MF Wang, R Ghosh, B Chaudhury, MY Kim · NeurIPS 2025 Workshop paper

  6. 2024

    Improving Instruction Following in Language Models through Proxy-Based Uncertainty Estimation

    J Lee, JO Woo, J Seok, P Hassanzadeh, W Jang · ICML 2024 ICML arXiv Code

  7. 2024

    Bayesian Active Learning for Semantic Segmentation

    S Didari, W Hu, JO Woo, H Hao, H Moon, S Min · Preprint arXiv

  8. 2023

    Unsupervised Accuracy Estimation of Deep Visual Models using Domain-Adaptive Adversarial Perturbation without Source Samples

    JH Lee, JO Woo, H Moon, K Lee · ICCV 2023 ICCV arXiv

  9. 2023

    Active Learning in Bayesian Neural Networks with Balanced Entropy Learning Principle

    JO Woo · ICLR 2023 ICLR arXiv Code

  10. 2023

    Unsupervised Contrastive Representation Learning for 3D Mesh Segmentation

    A Haque, H Moon, H Hao, S Didari, JO Woo, P Bangert · AAAI 2023

  11. 2022

    Analytic Mutual Information in Bayesian Neural Networks

    JO Woo · IEEE ISIT 2022 IEEE arXiv

  12. 2022

    Medical Image Labeling via Active Learning is 90% Effective

    P Bangert, H Moon, JO Woo, S Didari, H Hao · FICC 2022

  13. 2022

    Verifying Measures of Quantum Entropy

    GC Pastor, JO Woo · Applied Mathematics

  14. 2021

    Active Learning Performance in Labeling Radiology Images is 90% Effective

    P Bangert, H Moon, JO Woo, S Didari, H Hao · Frontiers in Radiology

  15. 2021

    Highly Efficient Representation and Active Learning Framework for Imbalanced Medical Image Classification

    H Hao, H Moon, S Didari, JO Woo, P Bangert · NeurIPS Workshop 2021

  16. 2021

    PatchNet: Unsupervised Object Discovery Based on Patch Embedding

    H Moon, H Hao, S Didari, JO Woo, P Bangert · Preprint

  17. 2019

    Majorization and Rényi Entropy Inequalities via Sperner Theory

    M Madiman, L Wang, JO Woo · Discrete Mathematics

  18. 2018

    On the Steady State of Continuous Time Stochastic Opinion Dynamics

    F Baccelli, S Vishwanath, JO Woo · Journal of Applied Probability

  19. 2018

    Entropy Inequalities for Sums in Prime Cyclic Groups

    M Madiman, L Wang, JO Woo · SIAM J. Discrete Mathematics

  20. 2017

    An Analytical Framework for Modeling a Spatially Repulsive Cellular Network

    CS Choi, JO Woo, JG Andrews · IEEE Trans. Communications

  21. 2017

    On the Coverage Probability of a Spatially Correlated Network

    CS Choi, JO Woo, JG Andrews · IEEE ISIT 2017

  22. 2016

    On the Entropy and Mutual Information of Point Processes

    F Baccelli, JO Woo · IEEE ISIT 2016

  23. 2015

    A Discrete Entropy Power Inequality for Uniform Distributions

    JO Woo, M Madiman · IEEE ISIT 2015

  24. 2015

    Information Theoretic Inequalities, Limit Theorems, and Universal Compression over Unknown Alphabets

    JO Woo · Ph.D. Dissertation, Yale University

  25. 2014

    A Lower Bound on the Rényi Entropy of Convolutions in the Integers

    L Wang, JO Woo, M Madiman · IEEE ISIT 2014

  26. 2014

    Redundancy of Exchangeable Estimators

    NP Santhanam, AD Sarwate, JO Woo · Entropy

  27. 2009

    A Note on Preconditioning by Low-Stretch Spanning Trees

    DA Spielman, JO Woo · Preprint

Open to research collaborations and invited talks on reliable reasoning, uncertainty, and trustworthy AI systems.

jaeoh.woo@aya.yale.edu →