yizhang.io Email

founder & ai scientist — new york

Yi Zhang

I build AI systems that learn continuously — and act optimally.

Co-founder & CTO of Greenshoe. Previously at AWS AI Labs, where I trained Amazon Titan models and pioneered the memory and RAG systems behind Amazon Bedrock and Amazon Bedrock AgentCore. PhD in computer science from the University of Pennsylvania.

Portrait of Yi Zhang
2025 — now

Building Greenshoe

Greenshoe is the AI-native platform for SEC disclosures: it drafts, benchmarks, and reviews 10-Ks, 10-Qs, and IPO filings — weeks of manual work compressed into hours — and continuously monitors market and regulatory signals so public companies and their law firms act in real time, not at the deadline. It's my research thesis in production: a system that learns continuously and acts optimally.

$3M seed led by AIX Ventures. We're hiring — talk to me.

research

Intelligence is an optimization process — over experience, compute, and data.

Most AI systems stop improving the day they ship: experience is thrown away, reasoning is recomputed from scratch, data goes unread. My research treats each as an optimization problem — learning from experience optimizes the model, planning optimizes the use of compute, and retrieval optimizes the knowledge every decision draws on.

post-trainingLearning from experience

Turning the knowledge, experience, and reflections an agent accumulates in memory into training signal — reinforcement learning that makes models and agents improve with use, from RLHF for grounded generation to agents that learn from their own past interactions.

Amazon Titan · Bedrock AgentCore Memory

test-time scalingFinding the optimal plan

Scaling reasoning at test time and deciding where compute goes — across steps, tools, and collaborating agents — while reusing prior reasoning so no plan starts from scratch.

Optimizable LLM Planning · FutureWeaver, COLM 2026 · Log-Augmented Generation, ICLR 2026

retrieval & dataRetrieval with reasoning

Retrieval that reasons over structured and unstructured data — alignment-oriented retrieval, join-aware multi-table retrieval, generative retrieval, and data-lake search — rooted in a decade of database systems research.

ARM, ACL 2025 · ACL 2024 Outstanding Paper · SIGMOD 2020

2011 — 2025

Experience

  1. 2025 — now

    Greenshoe, Inc. co-founder & cto

    AI research platform for SEC disclosures. New York.

  2. 2022 — 2025

    AWS AI Labs senior applied scientist

    • Led training of Amazon Titan foundation models for RAG — SFT and RLHF for query and grounded response generation.
    • Built advanced RAG for Amazon Bedrock Knowledge Bases: query rewriting, decomposition, and planning for complex information-seeking tasks.
    • Pioneered agent memory at AWS — early work behind Amazon Bedrock AgentCore Memory, letting agents learn from past interactions.
  3. 2018 — 2021

    Research internships

    Google Research ('21) · Microsoft Research ('20) · Microsoft CISL ('19) · IBM Research Almaden ('18)

  4. 2016 — 2022

    University of Pennsylvania phd, computer & information science

    Advised by Zachary G. Ives and Dan Roth.

  5. 2011 — 2016

    Fudan University bs, computer science

2016 — 2026

Selected publications

  • COLM 2026

    FutureWeaver: Planning Test-Time Compute for Multi-Agent Systems with Modularized Collaboration

    Dongwon Jung, Peng Shi, Muhao Chen, Yi Zhang

  • ICLR 2026

    Log-Augmented Generation: Scaling Test-Time Reasoning with Reusable Computation

    Peter Baile Chen, Yi Zhang, Dan Roth, Samuel Madden, Jacob Andreas, Michael Cafarella

  • ACL 2025

    Can We Retrieve Everything All at Once? ARM: An Alignment-Oriented LLM-based Retrieval Method

    Peter Baile Chen, Yi Zhang, Michael Cafarella, Dan Roth

  • ACL 2025

    On Synthetic Data Strategies for Domain-Specific Generative Retrieval

    Haoyang Wen, Jiang Guo, Yi Zhang, Jiarong Jiang, Zhiguo Wang

  • ACL 2024

    Is Table Retrieval a Solved Problem? Exploring Join-Aware Multi-Table Retrieval

    Peter Baile Chen, Yi Zhang, Dan Roth

    ★ Outstanding Paper — Towards Knowledgeable LMs @ ACL 2024

  • NAACL 2021

    Learning to Decompose and Organize Complex Tasks

    Yi Zhang, Sujay Jauhar, Julia Kiseleva, Ryen White, Dan Roth

  • SIGMOD 2020

    Finding Related Tables in Data Lakes for Interactive Data Science

    Yi Zhang, Zachary G. Ives

  • VLDB 2019

    Juneau: Data Lake Management for Jupyter

    Yi Zhang, Zachary G. Ives

    ★ Best Demonstration Award

Google Scholar →