Xiaochen Ma

Xiaochen Ma

Ph.D. Student in Computer Science and Engineering

HKUST

Biography

Xiaochen Ma (马晓晨)

I am a Ph.D. student in the Department of Computer Science and Engineering at the Hong Kong University of Science and Technology (HKUST), advised by Prof. Binhang Yuan and Prof. Wentao Zhang. My research focuses on the systems layer between heterogeneous raw data and large language model training: making data preparation easier to compose, distribute, reproduce, and operate with Ray and Apache Spark.

I am currently a Qingyun Program Research Intern with Tencent Hunyuan’s Foundation Model Department in Beijing, where I work on data infrastructure for large-scale LLM data preparation, including operator management and distributed pipelines for deduplication, clustering, and document and video processing.

Before starting my Ph.D., I was a Research Assistant at Peking University from January 2025 to January 2026, where I worked with Prof. Wentao Zhang on LLM data preparation systems, including DataFlow. Earlier, I was a full-time Research Assistant at Sichuan University from July 2023 to June 2024, working with Prof. Jizhe Zhou on image manipulation detection and localization. This progression from reproducible vision benchmarks to general-purpose data preparation systems motivates my current work on scalable and reusable data workflows.

I received my B.Eng. in Computer Science and Technology from the College of Computer Science at Sichuan University in June 2023. My undergraduate thesis was recognized by the university as an Outstanding Undergraduate Thesis.

Research impact Google Scholar 787 citations h-index10 i10-index11
Open-source impact GitHub 9,568 stars

Representative Open-Source Projects

Project lead · Accepted at SIGMOD'27 LLM Data Infra Ray
Project

A composable data-pipeline and workflow system for LLM data preparation, spanning data generation, cleaning, evaluation, and orchestration.

My contribution

Project lead and core contributor, coordinating a team of more than 10 students across system design, implementation, code review, integration, and releases. My engineering contributions span pipeline compilation and execution, operator and registry design, storage, serving, CLI tooling, and RayOrch integration.

NeurIPS 2024 Spotlight Benchmark
Project

A modular training and evaluation codebase and comprehensive benchmark for image manipulation detection and localization.

My contribution

Joint first author, codebase designer, and coding lead. I designed the modular training and evaluation framework and have maintained its metrics, model zoo, CLI, packaging, releases, and tests.

Current focus Ray Pipeline Parallelism
Project

A Ray-powered acceleration layer that turns MinerU PDF-to-Markdown parsing into a scalable multi-GPU and multi-node data pipeline.

My contribution

Project initiator and core developer. I built the initial package and release workflow, then refactored the sequential stages into an asynchronous pipeline with cross-stage overlap, while maintaining benchmarks and compatibility.

First-author paper · arXiv'26 Ray Async Pipelines
Project

A programming model and distributed execution engine for foundation-model data preparation, preserving parent-child lineage, ordering, completion, and result assembly as each input expands into multiple records.

My contribution

First author and project initiator. I designed the dataflow model and runtime for ordered variable-cardinality expansion, compiler validation, and parent-scoped failure handling, and integrated the Ray-based executor into DataFlow.


Academic Service

Reviewer: AAAI'27, ICLR'27, NeurIPS'26, BMVC'26, ECCV'26, CVPR'26, ICLR'26, AISTATS'26, AAAI'25, NeurIPS'25, ICCV'25, ICML'25, CVPR'25, ICLR'25, AISTATS'25, NeurIPS'24, and ACM MM'24.

Interests
  • LLM Data Infrastructure
  • LLM Data Preparation and Workflows
  • High-Performance Distributed Systems
  • Ray and Apache Spark
Education
  • Ph.D in Computer Science and Engineering, 2028 (expected)

    HKUST

  • B.Eng. in Computer Science, 2023

    Sichuan University

Experience

Research & Industry Jan 2025 – Jan 2026
Full-time Research Assistant
China
Advisor: Prof. Wentao Zhang.
Research: Data preparation systems for large language models.
Research & Industry Jul 2023 – Jun 2024
Full-time Research Assistant
China
Advisor: Prof. Jizhe Zhou.
Research: Image manipulation detection and localization, including benchmark and open-source codebase development.
Education Sep 2019 – Jun 2023
Undergraduate Student
China
B.Eng. in Computer Science and Technology; Outstanding Undergraduate Thesis.
(2026). RayOrch: Programming and Executing Lineage-Controlled Multi-Grain Dataflows for Foundation-Model Data Preparation. arXiv'26.

PDF Code arXiv

(2024). IMDL-BenCo: A Comprehensive Benchmark and Codebase for Image Manipulation Detection & Localization. NeurIPS'24 Spotlight.

PDF Cite Code Poster arXiv NeurIPS URL

(2025). DataFlow: An LLM-Driven Framework for Unified Data Preparation and Workflow Automation in the Era of Data-Centric AI. ACM SIGMOD'27.

PDF Code arXiv

(2023). IML-ViT: Benchmarking Image Manipulation Localization by Vision Transformer. arXiv'23.

PDF Cite Code arXiv

(2026). DataFlow-Harness: A Grounded Code-Agent Platform for Constructing Editable LLM Data Pipelines. arXiv'26.

PDF Code arXiv

(2023). Pre-training-free Image Manipulation Localization through Non-Mutually Exclusive Contrastive Learning. ICCV'23.

PDF Cite Code arXiv URL

(2025). ForensicHub: A Unified Benchmark & Codebase for All-Domain Fake Image Detection and Localization. NeurIPS'25.

PDF Cite Code arXiv

(2024). Mesoscopic insights: Orchestrating multi-scale & hybrid architecture for image manipulation localization. AAAI'25.

PDF Cite Code DOI arXiv AAAI URL

Contact

  • xiaochen.ma.cs[at]gmail.com OR xiaochen.ma[at]connect.ust.hk
  • Kowloon, Hong Kong SAR,
  • The Hong Kong University of Science and Technology, Clear Water Bay, Kowloon, Hong Kong

Live Stats

Full view
View overview Last 28 days · Views · Sessions · Geography
Loading live statistics…