AI researcher & engineer

Reliable AI,
from theory
to practice.

I’m Wuxinlin. I connect mathematical foundations with machine learning to build systems that are more robust, efficient, and useful in the real world.

Currently at KLA AI Algorithm Engineer

ICCAD 2026 TPC member & session chair
STRUCTURE → LEARNING → SYSTEMS
A shaded mathematical manifold with fine mesh, contour traces, and connected nodes around an ochre focal point.
A mathematical lens.
A practical purpose.
A conceptual illustration of
the connections behind my work.
01

Robust learningStability under changing conditions

02

Efficient intelligenceBetter use of models and data

03

Applied AI systemsFrom industrial vision to avatars

01 / Selected work

Ideas, put to work.

Research and engineering across model reliability,
industrial perception, and generative systems.

Illustrated wafer particles and their corresponding segmentation masks, with matching locations and boundaries.
Image to particle maskIllustration
Industrial AIKLA · 2024–present

Tiny-particle wafer segmentation

Structure-aware segmentation for tiny-particle wafer inspection, bringing together boundary information, local complexity, and a compact student model.

Computer visionKnowledge distillation
Inside the work

The challenge. Preserve fine structural detail while keeping a vision model compact enough for high-throughput inspection.

The approach. A teacher–student pipeline that uses boundary salience, structural complexity, and local feature variation to guide learning.

The result. A 7 percentage-point mIoU improvement to 80% in internal validation of the prototype.

About the cover. The cover illustrates image-to-mask segmentation; it is not a KLA inspection sample or model output.

Random noise and conditioning inputs pass through a diffusion model to create illustrative spectral samples.
Noise to synthetic spectraMethod illustration
Industrial / Generative AIKLA

Conditional diffusion for metrology

Conditional diffusion for spectral-data synthesis, exploring how generative modeling can improve metrology under distribution shift.

Diffusion modelsOOD robustness
Inside the work

The challenge. Build more reliable multi-label regression models from noisy, high-dimensional physical measurements.

The approach. Synthesize spectral data with conditional diffusion and evaluate both in-distribution and out-of-distribution behavior.

The context. Industry research at KLA. The implementation and internal evaluation data are not shared here.

About the cover. The signals illustrate conditional generation. They are not industrial measurements or model evaluation results.

A circuit schematic with sensitive pins highlighted, beside a published DES comparison of predicted delay changes.
CirSTAG / DES benchmarkPublished data
ResearchDAC 2025

CirSTAG: circuit-model sensitivity

CirSTAG studies circuit-model stability on graph-based manifolds, connecting spectral methods with the behavior of graph neural networks.

Graph learningBest Paper Nominee
Inside the work

The question. How can we understand and improve the stability of learned models for circuit design?

The approach. Analyze input and output graph-based manifolds to identify sensitive circuit elements under changes in graph structure and node features.

The outcome. First-author research at DAC 2025, selected as a Best Paper Nominee.

Read CirSTAG

About the cover. The circuit is schematic. The DES bars replot Table I of CirSTAG: mean relative change in GNN-predicted arrival time, with 10% of nodes perturbed and pin capacitance scaled by 10. Stable: 0.0068%; sensitive: 0.5211%.

Handwritten digit examples from the SPADE project figure pass through a schematic sensitivity analysis to identify fragile samples.
SPADE / handwritten digitsSource examples
ResearchICML 2021

SPADE: black-box robustness

SPADE evaluates black-box adversarial robustness through the relationship between input and output graph manifolds.

Spectral methodsAdversarial robustness
Inside the work

The question. How can we assess model and sample sensitivity without relying on access to a model’s internal gradients?

The approach. Use a spectral graph-based measure to study local input–output behavior and inform robustness evaluation and adversarial training.

Read SPADE

About the cover. The digit tiles come from the original SPADE project figure. The connecting graph illustrates the method; it does not assign new numerical scores or show clean/adversarial image pairs.

Illustrative text variants pass through a language model into a graph comparison that highlights a sensitive relationship.
SALMAN / language-model stabilityMethod illustration
ResearchPreprint · 2025

SALMAN: language-model stability

A graph-manifold approach to finding language-model inputs whose small changes can produce unstable output behavior.

Language modelsRobustness evaluation
Inside the work

The question. Which text samples are most sensitive to perturbations?

The approach. Compare input and output relationships using Distance Mapping Distortion, then use sample-level stability to guide robustness evaluation and training.

The status. Research preprint; first author.

Read the preprint

About the cover. A conceptual view of input–output relationship distortion, not actual model responses or measured stability scores.

Corresponding graph neighborhoods show an edge change and an altered output relationship, highlighting sensitivity in a graph neural network.
SAGMAN / graph-model stabilityMethod illustration
ResearchPreprint · 2024

SAGMAN: graph-model stability

Spectral analysis of how graph neural networks transform relationships between nodes, revealing sensitivity to changes in edges and features.

Graph neural networksSpectral methods
Inside the work

The question. Which graph nodes become unstable when connections or features change?

The approach. Build distance-preserving graph-based manifolds with spectral embeddings and probabilistic graphical models, then analyze distortion between model input and output.

The status. Research preprint; first author. Applications include GNN stability, recommendation models, and robustness research.

Read the preprint

About the cover. A conceptual graph perturbation and mapping. Highlighted connections illustrate sensitivity; they are not a measured benchmark result.

Sampling a flow domain around an obstacle, followed by an actual absolute-pressure-error field from the SGM-PINN paper.
SGM-PINN / fluid simulationPaper result
ResearchDAC 2024

SGM-PINN: efficient physics learning

SGM-PINN uses sampling graphical models to make training physics-informed neural networks more efficient.

Scientific MLTraining efficiency
Inside the work

The question. Can a model learn the relevant physics while spending less time on redundant training samples?

The approach. Graph-based importance sampling for PINNs, studied on computational fluid dynamics workloads.

My contribution. Co-author on the DAC 2024 publication.

Read SGM-PINN

About the cover. Sampling locations are illustrative. The pressure-error panel is adapted from Anticev, Aghdaei, Cheng & Feng, SGM-PINN, Figure 4; its field colors and color scale are preserved. This adapted cover is licensed CC BY-NC-SA 4.0.

A conceptual copper cross-section shows a recessed surface and a dishing-depth measurement, connected to an ML estimation workflow.
Copper dishing / learning to measureMethod illustration
Industrial AISPIE presentation · 2026

Learning to measure copper dishing

Machine learning for Cu dishing metrology in advanced packaging, connecting semiconductor measurements with data-driven estimation.

Semiconductor metrologyApplied ML
Inside the work

The work. Developed ML methods for semiconductor metrology and advanced packaging at KLA.

The presentation. Machine-Learning Solution for Cu Dishing Metrology in Advanced Packaging, SPIE Advanced Lithography + Patterning 2026.

The context. Industry research; proprietary implementation and evaluation data are not shared here.

About the cover. A schematic cross-section and measurement workflow. Geometry is conceptual, with no measured dimensions or proprietary sample data.

Three actual talking-avatar demo frames show the same character with changing mouth and facial expressions.
RITA / talking-avatar demoActual demo frames
Generative AICVPR 2024 demo

RITA: real-time talking avatars

Real-time talking avatars that connect facial reconstruction, speech synthesis, lip synchronization, and interactive dialogue.

Multimodal systemsReal-time inference
Inside the work

The work. As a scientific advisor and AI-lab lead, I worked on 3D avatar assistants combining visual, speech, emotion, and language components.

The demonstration. Co-developed RITA, a real-time interactive talking-avatar framework demonstrated at CVPR 2024.

View the CVPR demo program (PDF)

About the cover. The cover uses actual output frames at 0.0, 1.0, and 3.0 seconds from the project demo. These are not input portraits.

Actual vial inspection images show negative and positive source examples, with original annotations retained.
Vial classification / inspection examplesSource examples
Industrial AILEMA · 2021–2023

Efficient vial inspection

Industrial vial classification that brings together data curation, model pruning, and robust training to improve inspection quality and throughput.

Computer visionModel compression
Inside the work

The challenge. Improve classification while reducing inference cost in an industrial inspection pipeline.

The approach. Curate the training data, prune the network, and strengthen training against perturbations.

The result. A 42% relative accuracy improvement over the prior model and a 5.8× inference speedup in internal evaluation.

About the cover. The negative and positive examples come from the original LEMA project figure. Boxes and arrows are source annotations; these are not before-and-after model outputs.

An actual material-inspection image and an enlarged crop reveal the same fine hair-like contaminant inside its original annotation.
Hair detection / overview and detailSource example
Industrial AILEMA · 2021–2023

Finding fine contaminants

Hair detection in visually complex inspection images, with an emphasis on learning from limited data and maintaining reliability under image corruption.

Small-data learningRobust vision
Inside the work

The challenge. Detect fine hair-like contaminants against complex material textures.

The approach. Robust training and augmentation, evaluated on a small internal dataset and severity-5 image corruptions.

The context. Scientific advisory work on industrial computer vision at LEMA.

About the cover. The overview and magnified detail use the same original inspection image. The red box is a source annotation, not a new prediction.

A conceptual pipeline separates historical observations at a forecast cutoff, combines specialist estimates, and produces a calibrated forecast.
Past observations to calibrated forecastsMethod illustration
ML systemsIndependent research · 2026

Forecasting with calibrated uncertainty

A point-in-time forecasting and model-comparison system that combines specialist predictions, calibrated probabilities, and tests of ranking stability.

Probabilistic forecastingModel evaluation
Inside the work

The question. How should models be compared when predictions arrive at different times and evidence is incomplete?

The approach. Split data by label-resolution date, combine specialist models with contextual log-opinion pooling, calibrate outputs, and compare matched forecasts with Bradley–Terry/Elo rankings.

The scope. Independent research using daily equity data, reproducible artifacts, and offline evaluation.

About the cover. A conceptual forecasting workflow. The drawn observations and forecast are illustrative, not measured predictions, performance results, or investment returns.

02 / Publications & presentations

The research behind the work.

Selected contributions to machine learning,
design automation, and applied AI.

  1. 2026
    SPIE Advanced Lithography + PatterningPresentation

    Machine-Learning Solution for Cu Dishing Metrology in Advanced Packaging

    Semiconductor metrology · Conference presentation

    Industry research
  2. 2025
    DACBest Paper Nominee

    CirSTAG: Circuit Stability Analysis on Graph-based Manifolds

    First author · Graph learning & stability

    Paper (PDF)
  3. 2024arXiv
  4. 2024
    CVPRDemo

    RITA: A Real-Time Interactive Talking Avatars Framework

    Co-developer · Multimodal generative systems

    Program (PDF)
  5. 2021
    ICMLPaper

    SPADE: A Spectral Method for Black-Box Adversarial Robustness Evaluation

    Co-first author · Adversarial robustness

    Proceedings

03 / A little background

A researcher’s curiosity.
An engineer’s mindset.

I’m an AI researcher and engineer with a Ph.D. in Computer Engineering from Stevens Institute of Technology. My work bridges the mathematical foundations of machine learning and the demands of real systems.

At KLA, I develop AI for semiconductor metrology and inspection. Previously, I researched adversarial robustness and graph learning at Stevens, advised industrial vision teams, and led an AI lab building interactive avatar systems.

The thread through it all: understand why a model works, where it fails, and how to make it useful.

Wuxinlin Cheng standing outdoors in front of a castle.
Wuxinlin ChengResearcher. Engineer. Always curious.

Along the way

  1. 2024–present

    KLA Corporation

    AI Algorithm Engineer, Deep Learning

    Semiconductor metrology & inspection
  2. 2020–2025

    Stevens Institute of Technology

    Research Assistant

    Adversarial robustness & graph learning
  3. 2023–2024

    Stealth Startup

    Scientific Advisor & AI Lab Lead

    Multimodal avatars · Concurrent with Ph.D.
  4. 2021–2023

    LEMA Technology

    Scientific Advisor

    Industrial computer vision · Concurrent with Ph.D.

Professional service / 2026

ICCAD 2026

International Conference on Computer-Aided Design

Invited Technical Program Committee (TPC) Member

AI Algorithms and Applications track

Session Chair

Chairing two sessions:

  • Foundation Model Benchmarks
  • Trustworthy AI for Hardware Design

Foundations

  • Ph.D., Computer EngineeringStevens Institute of Technology · 2025
  • M.Eng., Computer EngineeringStevens Institute of Technology · 2020
  • B.Eng., Electrical EngineeringSichuan University · 2018

Recognition & service

  • 2026

    Global top-three winning team & challenge-track winnerKLA CY26 AI/Algorithm Hackathon

  • 2025

    Outstanding Ph.D. Dissertation AwardComputer Engineering · Stevens Institute of Technology

  • Service

    ICLR 2026 reviewer

  • Invention

    Named inventor on two KLA internal invention disclosures

Tools I work with

Python / C++ / PyTorch / TensorFlow / Scientific computing

04 / Let’s connect

Good work starts
with a conversation.

For research collaborations, applied AI opportunities,
or a thoughtful exchange of ideas.

Project cover

Project cover, enlarged.