<?xml version="1.0" encoding="UTF-8"?><rss xmlns:dc="http://purl.org/dc/elements/1.1/" xmlns:content="http://purl.org/rss/1.0/modules/content/" xmlns:atom="http://www.w3.org/2005/Atom" version="2.0" xmlns:cc="http://cyber.law.harvard.edu/rss/creativeCommonsRssModule.html">
    <channel>
        <title><![CDATA[Stories by Daniel Kang on Medium]]></title>
        <description><![CDATA[Stories by Daniel Kang on Medium]]></description>
        <link>https://medium.com/@danieldkang?source=rss-15128583ec9e------2</link>
        <image>
            <url>https://cdn-images-1.medium.com/fit/c/150/150/1*8sAXYMCZGbP8K09lihLN-Q.jpeg</url>
            <title>Stories by Daniel Kang on Medium</title>
            <link>https://medium.com/@danieldkang?source=rss-15128583ec9e------2</link>
        </image>
        <generator>Medium</generator>
        <lastBuildDate>Sun, 15 Nov 2026 04:18:00 GMT</lastBuildDate>
        <atom:link href="https://medium.com/@danieldkang/feed" rel="self" type="application/rss+xml"/>
        <webMaster><![CDATA[yourfriends@medium.com]]></webMaster>
        <atom:link href="http://medium.superfeedr.com" rel="hub"/>
        <item>
            <title><![CDATA[Noisy Data Breaks RLVR]]></title>
            <link>https://medium.com/@danieldkang/noisy-data-breaks-rlvr-7639b7277ec3?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/7639b7277ec3</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Mon, 03 Aug 2026 17:42:18 GMT</pubDate>
            <atom:updated>2026-08-03T17:42:18.368Z</atom:updated>
            <content:encoded><![CDATA[<p><a href="https://arxiv.org/abs/2402.03300">Reinforcement learning with verifiable rewards</a> (RLVR) is a widely used post-training paradigm to improve the reasoning capabilities of LLMs. However, creating the high-quality verifiable answers needed to train the model is labor-intensive and expensive, as highlighted by the boom of data labeling companies, such as <a href="https://www.linkedin.com/posts/brendan-foody-2995ab10b_2025-wrapped-mercor-grew-revenue-by-4658-activity-7412326886686507008-sMY-/">Mercor</a>. This creates a fundamental question for post-training: <em>To what extent can RLVR, with algorithmic improvements, tolerate large-volume, noisy data?</em></p><p><a href="https://arxiv.org/pdf/2506.10947">Recent</a> <a href="https://arxiv.org/pdf/2505.22653">literature</a> promotes a counter-intuitive idea: RLVR is robust to noisy data. <a href="https://arxiv.org/pdf/2506.10947">Prior work</a> shows that training on “100% incorrect” answers leads to only 5% lower performance than clean data and achieves higher performance than format rewards (enclosing <em>any</em> answer in \boxed{}). This suggests we can throw cheap, messy annotations at an LLM, tweak the <a href="https://arxiv.org/pdf/2509.26114">RLVR loss</a>, and still significantly improve reasoning.</p><p>We find that this hypothesis is false. The “surprisingly” high effectiveness of RLVR on “100%” noisy data is due to the contamination in the synthetic noise — the claimed “100%” noisy data contains a significant portion of clean data. In this blog, we develop a more rigorous noisy data curation pipeline and show that noisy data is destructive to RLVR, impacting the test accuracy by over 9%. Even current advanced algorithmic improvements fail to mitigate the severe impact.</p><p>Paper: <a href="https://arxiv.org/abs/2603.16140">https://arxiv.org/abs/2603.16140</a></p><p>Repo: <a href="https://github.com/uiuc-kang-lab/rlvr-noisy-data">https://github.com/uiuc-kang-lab/rlvr-noisy-data</a></p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*7AgL9bbYNWKVgLqD.png" /><figcaption><em>Training on noisy data leads to significantly degraded performance compared to training on clean data.</em></figcaption></figure><h3>Prior “noisy” data contains correct labels</h3><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*erzzEn7EoqcI6gDQ.png" /><figcaption><em>Examples of correct labels contained in the prior “noisy” data.</em></figcaption></figure><p>Recent work claimed that <a href="https://arxiv.org/pdf/2506.10947">LLMs could learn effectively from 100% incorrect annotations during RLVR</a>. But when we verified the synthetic noisy datasets used in prior work, we found a critical flaw: the noise was contaminated with correct answers. This hidden contamination inflated the performance of models trained on it.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*BJEomD5em0XjNgXR.png" /><figcaption><em>Data re-verification pipeline for synthesizing a truly noisy dataset.</em></figcaption></figure><p>How did this happen? Prior work generated noisy labels for math datasets by sampling answers from a base LLM and filtering out the correct ones using basic symbolic equivalence checkers. However, this filtering fails for two primary reasons:</p><ol><li><strong>Insufficient gold answers</strong>: A math problem often has multiple valid solutions, but ground-truth annotations typically only capture one (as shown in Example 1 above).</li><li><strong>Inadequate equivalence checking</strong>: Basic symbolic verifiers frequently fail to recognize when a generated answer is mathematically identical to the gold answer but formatted differently (as shown in Example 2 above).</li></ol><p>To measure the <em>actual</em> impact of noise, we must ensure the noisy data has incorrect answers. We applied a more rigorous re-verification pipeline using GPT-5 Pro combined with manual expert review. We discovered that 16% of the supposedly “incorrect” labels in the original dataset were actually correct<strong>.</strong> By purging these leaked answers, we constructed a <em>truly</em> noisy dataset.</p><h3>The collapse of reasoning</h3><p>When we purged those hidden correct annotations, the performance collapsed. In our experiments using Qwen2.5-Math-7B, training on 100% truly incorrect annotations dropped average accuracy on five math benchmarks (AIME’24, AIME’25, AMC’23, AMC’24, and MATH500) by 9% compared to training on clean data. This performance was even worse than training with format-only rewards.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*50oHudNFA8AQHJ9q.png" /><figcaption><em>Training on noisy data leads to significantly degraded performance compared to training on clean data.</em></figcaption></figure><p>We also found that noisy training data leads to weaker reasoning. When scaling the number of attempts (k &gt; 1), pass@k dropped below the base model’s baseline capabilities. Furthermore, noise penalizes deep exploration. Models trained on noisy data produced reasoning chains that were 5–24% shorter than those trained on clean data.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*UARKyz0-PiWV84_g.png" /><figcaption><em>Noise leads to weaker reasoning (lower Pass@k and shorter response length).</em></figcaption></figure><h3>Existing algorithmic improvements fail on noisy data</h3><p>Prior research has proposed a wide range of algorithmic improvements to the vanilla GRPO. Can these existing algorithm improvements make RLVR more robust to noisy data? Unfortunately, our empirical results show that they cannot.</p><p>We considered state-of-the-art RLVR variants: <a href="https://arxiv.org/pdf/2511.20347v1">SAPO</a>, <a href="https://arxiv.org/abs/2503.14476">DAPO</a>, <a href="https://fengyao.notion.site/off-policy-rl">TIS</a>, <a href="https://arxiv.org/abs/2503.20783">DR. GRPO</a>, and <a href="https://arxiv.org/pdf/2510.00915v3">PGFC</a> (an algorithm explicitly designed to mitigate the impact of noise). We tested them under a 50% noise rate, a reasonable rate identified on <a href="https://arxiv.org/abs/2601.08778">real-world text-to-SQL datasets</a>. Unfortunately, all of these algorithms lead to more than 5.0% accuracy degradation on at least one benchmark compared to training with clean data. On AIME and AMC benchmarks, these algorithms achieve comparable or lower performance than training with only format rewards.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*dqRZvkpTfLzCq0Cc.png" /><figcaption><em>Under 50% noise, all of the improved algorithms lead to &gt; 5% accuracy degradation on at least one benchmark compared to training with clean data.</em></figcaption></figure><h3>Conclusion</h3><p>Our empirical evidence shows that the impact of noisy data is destructive and fundamental to RLVR. Algorithmic tweaks simply cannot yet compensate for performance loss due to noisy training data. If we want to improve LLM reasoning, high-quality data remains essential.</p><p>We released our validated noisy dataset, model checkpoints, and training scripts at: <a href="https://github.com/uiuc-kang-lab/rlvr-noisy-data">https://github.com/uiuc-kang-lab/rlvr-noisy-data</a>. For more technical details, please refer to our paper: <a href="https://arxiv.org/abs/2603.16140">https://arxiv.org/abs/2603.16140</a>.</p><p><em>Written by Yuxuan Zhu and Daniel Kang</em></p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=7639b7277ec3" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[Pervasive Annotation Errors Break Text-to-SQL Benchmarks and Leaderboards]]></title>
            <link>https://medium.com/@danieldkang/pervasive-annotation-errors-break-text-to-sql-benchmarks-and-leaderboards-9fba58b79fc1?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/9fba58b79fc1</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Mon, 27 Jul 2026 18:08:41 GMT</pubDate>
            <atom:updated>2026-07-27T18:08:41.589Z</atom:updated>
            <content:encoded><![CDATA[<p><a href="https://cloud.google.com/blog/products/databases/techniques-for-improving-text-to-sql">Text-to-SQL</a> is increasingly important in data analytics and database-driven applications. It reduces the effort required for experienced users to write complex SQL queries while enabling non-SQL users to query databases through natural language. Reflecting this growing demand, major data platforms now support text-to-SQL capabilities through systems such as <a href="https://www.databricks.com/product/genie/agents">Databricks Genie Agents</a>, <a href="https://www.snowflake.com/en/product/features/cortex/">Snowflake Cortex</a>, <a href="https://docs.cloud.google.com/bigquery/docs/write-sql-gemini">Gemini in BigQuery</a>, and <a href="https://aws.amazon.com/quick/quicksight/">Amazon Quick Sight</a>.</p><p>To support the development and evaluation of text-to-SQL systems, researchers have developed a growing number of benchmarks. However, <strong>we found pervasive annotation errors</strong> in two widely used text-to-SQL benchmarks: <a href="https://bird-bench.github.io/">BIRD</a> and <a href="https://spider2-sql.github.io/">Spider 2.0-Snow</a>.</p><p>Do these errors change how we judge the state of the art in text-to-SQL agents? To answer this question, we re-evaluated all 16 open-source agents from the BIRD leaderboard, uncovering relative performance changes ranging from −7% to +31% and ranking shifts of -9 to +9 positions. The previous state-of-the-art, <a href="https://contextual.ai/blog/open-sourcing-the-best-local-text-to-sql-system">Contextual-SQL</a>, dropped from 1st to 7th, while <a href="https://arxiv.org/abs/2505.14174">GenaSQL</a> (6th) and <a href="https://arxiv.org/pdf/2405.16755">CHESS</a> (7th) rose to a tie for 1st. These findings show that annotation errors can significantly distort reported performance and rankings, underscoring the urgent need for high-quality text-to-SQL benchmarks. And if you are an agent or model developer, don’t blindly trust benchmark scores or leaderboards! Dive deeper into our findings in our latest <a href="https://arxiv.org/abs/2601.08778">paper</a>.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*jCj1iddjqtaqcKUX.png" /><figcaption>Agent execution accuracy and ranking changes from the original to the corrected BIRD Dev subsets.</figcaption></figure><h3>Annotation Errors Are Pervasive in Text-to-SQL Benchmarks</h3><p>We conducted an in-depth analysis of annotation errors in <a href="https://bird-bench.github.io/">BIRD</a> and <a href="https://spider2-sql.github.io/">Spider 2.0-Snow</a>, two benchmarks widely used by both academic and industrial teams (e.g., <a href="https://arxiv.org/abs/2410.01943">Google</a>, <a href="https://openai.com/index/gpt-4o-fine-tuning/">OpenAI</a>, <a href="https://www.arxiv.org/pdf/2509.21459">Databricks</a>, <a href="https://www.snowflake.com/en/engineering-blog/real-time-text-to-sql-snowflake-intelligence/">Snowflake</a>, <a href="https://www.notion.so/Q-SQL-AWS-2bfe8bf30514801ba112f92ff20784e3">Amazon</a>, <a href="https://arxiv.org/abs/2507.04701">Alibaba</a>, and <a href="https://arxiv.org/abs/2503.02240">ByteDance</a>).</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/854/0*iOFA9TgoMM-altft.png" /><figcaption>BIRD has the largest number of evaluated models (100) on its leaderboard, while Spider 2.0-Snow has the highest average token count of its ground-truth SQL queries (data cutoff: October 20, 2025).</figcaption></figure><p>Our analysis consists of three stages: (1) an AI agent producing per-annotation diagnostic reports; (2) human SQL experts adjudicating all cases flagged by the agent and assigning final correctness labels; and (3) to identify errors the agent may have missed, the experts also review unflagged examples for recurring error patterns uncovered in stage two.</p><p>Through our in-depth examination, we found that:</p><ul><li><a href="https://github.com/bird-bench/mini_dev">BIRD Mini-Dev</a> has an annotation error rate of 52.8%</li><li><a href="https://arxiv.org/pdf/2411.07763">Spider 2.0-Snow</a> (August 20, 2025 date cutoff) has an annotation error rate of 62.8%</li></ul><p>In the rest of this blog post, we first assess the impact of these errors, then break them down into four recurring patterns with illustrative examples, and finally introduce our AI agent we designed to assist finding these errors.</p><h3>Annotation Errors Can Significantly Change Measured Agent Performance and Leaderboard Rankings</h3><p>We evaluated all open-source text-to-SQL agents listed on the <a href="https://bird-bench.github.io/">BIRD leaderboard</a> as of August 20, 2025. To quantify the impact of annotation errors, we randomly sampled 100 examples from the BIRD Dev set and manually corrected all annotation issues we identified. We then re-evaluated all 16 open-source agents on both the original subset and the corrected version.</p><p>After re-evaluation, we found</p><ul><li>Agent execution accuracy changes ranging from −4% to +19%</li><li>Agent ranking changes ranging from −9 to +9 positions</li></ul><p>Notably, <a href="https://github.com/ShayanTalaei/CHESS">CHESS</a> and <a href="https://github.com/genaasia/N-rep">GenaSQL</a> demonstrate large relative improvements (+30.6% and +26.6%, respectively), moving from mid-ranked positions (7th and 6th) to a tie for 1st place at 81% execution accuracy on the corrected subset.</p><p>These results demonstrate that the measured performance and rankings of agents are highly sensitive to the annotation errors in the benchmark. When error rates are high, the reliability of the benchmark and leaderboard deteriorates, and it may no longer reflect the true relative performance of different agents.</p><p><strong>Takeaway:</strong> Our findings highlight the urgent need for high-quality text-to-SQL benchmarks. And if you’re developing or selecting a text-to-SQL agent for a downstream application, don’t blindly rely on existing benchmark scores or leaderboards — verify the data and inspect failure cases.</p><h3>Where Do These Errors Come From?</h3><p>A text-to-SQL annotation instance consists of three components: the natural language input (T), the ground-truth SQL query (Q), and the target database (D). Annotation errors can occur within any of these components or arise from inconsistencies among them. We categorize such errors into four recurring patterns:</p><ul><li>E1: Mismatches between the semantics of <em>Q</em> and the logic of <em>T</em>.</li></ul><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*W6ltRzvmN7kHynA3.png" /></figure><ul><li>E2: Mismatches between the semantics of <em>Q</em> and <em>D</em> due to limited understanding of the data and the schema.</li></ul><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*As0mq9m3NyvztPyY.png" /></figure><ul><li>E3: Mismatches between the semantics of <em>Q</em> and the domain knowledge relevant to <em>T</em>, or misannotated domain knowledge in <em>T</em>.</li></ul><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*J6UNL9OSkQnJiOBh.png" /></figure><ul><li>E4: Ambiguity in <em>T</em>, such as multiple possible interpretations or an unclear output format.</li></ul><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*xt8cN45G-Q3EP6z8.png" /></figure><p>We categorized examples by error pattern (each example may exhibit multiple patterns). E2 is the most common in both benchmarks: among examples with annotation errors, it occurs in 57.79% of BIRD Mini-Dev examples and 57.89% of Spider 2.0-Snow examples.</p><h3>Human-in-the-Loop Error Detection</h3><p>To make error detection scalable and systematic, we introduced SAR-Agent (SQL Annotation Reviewer), to our knowledge, the first AI agent designed to assist SQL experts in detecting text-to-SQL annotation errors.</p><h3>SAR-Agent: SQL Annotation Reviewer</h3><p>Given an annotation (the natural-language input and the “ground-truth” SQL) and the target database schema, SAR-Agent interacts with the database to generate a diagnostic report assessing the correctness of the annotation. Human SQL experts then review SAR-Agent-flagged cases, verify the stated reasons, and assign the final correctness label. To identify errors the agent may have missed, the experts also examine unflagged examples for recurring error patterns uncovered during this review.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*ntKOA8Z6MxoV53v_.png" /><figcaption>Overview of the SAR-Agent framework.</figcaption></figure><h3>SAR-Agent Helps Experts Find More Errors</h3><p>We evaluated SAR-Agent’s ability to detect annotation errors by comparing it against SQL experts from <a href="https://docs.google.com/spreadsheets/d/1IGm9Otruey60ujUnl8AOkepY3qgWHdFJHnX7hQGUeCw/edit?gid=0#gid=0">the Arcwise team</a>. After manual verification, we found that SAR-Agent correctly detects 138 of 161 Arcwise-reported error cases (85.7% hit rate). In addition, SAR-Agent identifies 41.6% more error examples than the Arcwise team.</p><p>We released Arcwise-Plat-SQL and Arcwise-Plat on our <a href="https://github.com/uiuc-kang-lab/text_to_sql_benchmarks/tree/main">GitHub repository</a>, extending Arcwise’s corrections to produce cleaner, more reliable BIRD Mini-Dev variants. We also made the <a href="https://github.com/uiuc-kang-lab/text_to_sql_benchmarks/tree/main/SAR-Agent">SAR-Agent code</a> publicly available.</p><h3>Conclusion</h3><p>In this work, we find that widely used text-to-SQL benchmarks contain high annotation error rates, which substantially undermine the reliability of benchmark results and leaderboards. We develop a toolkit and offer practical guidance for building high-quality text-to-SQL benchmarks.</p><p>To better support both text-to-SQL research and real-world deployment, benchmark developers should prioritize annotation quality — not just quantity. We advocate using our proposed pipeline to create more reliable text-to-SQL benchmarks. Please see our <a href="https://arxiv.org/abs/2601.08778">paper</a> for details and our <a href="https://github.com/uiuc-kang-lab/text_to_sql_benchmarks/tree/main">GitHub repository</a> for the released resources.</p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=9fba58b79fc1" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[Provable Generalization Bounds for RLVR at the Billion-Parameter Scale]]></title>
            <link>https://medium.com/@danieldkang/provable-generalization-bounds-for-rlvr-at-the-billion-parameter-scale-c3faa4d63967?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/c3faa4d63967</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Mon, 20 Jul 2026 18:16:12 GMT</pubDate>
            <atom:updated>2026-07-20T18:31:14.985Z</atom:updated>
            <content:encoded><![CDATA[<p>Reinforcement learning with verifiable rewards (RLVR) has become a critical tool for modern large language model (LLM) development. RLVR is used to improve the general-purpose reasoning abilities of frontier language models, and has proven particularly critical for <a href="https://arxiv.org/abs/2402.03300">mathematical problem solving</a> and <a href="https://arxiv.org/pdf/2603.24477">agentic coding</a>. Beyond the frontier, a model which is trained via RLVR can outperform much larger frontier models in specialized domains, making this approach especially attractive for enterprise tasks involving proprietary data. For example, <a href="https://www.databricks.com/blog/power-rlvr-training-leading-sql-reasoning-model-databricks">Databricks</a> uses RLVR to train a text-to-SQL specialist that outperforms much larger frontier models in this domain.</p><p>Unfortunately, this kind of specialization often comes at the risk of overfitting: it can be unclear whether a model trained via RLVR has learned to generalize in deployment, or if it is simply memorizing its training data. Empirical studies have reported mixed results [<a href="https://arxiv.org/pdf/2512.20760">1</a>, <a href="https://arxiv.org/pdf/2506.19733">2</a>, <a href="https://arxiv.org/pdf/2510.02230">3</a>], and our theoretical understanding of generalization remains limited in this context. Can we move beyond empirics to theoretically certify the generalizability of models trained via RLVR?</p><p>In this blog post, we introduce a framework that verifies the generalizability of RLVR models on fresh data drawn from the same distribution as the training data, and establish the first non-vacuous generalization bounds for RLVR at the billion-parameter scale. Across four real-world domains (math, programming, general-knowledge reasoning, and Text-to-SQL), our lower bounds for Qwen3.5–4B exceed the base model’s accuracy by 9–51%.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*SfnciAFDlpjKzg2v.png" /></figure><p>For more details, please check out our <a href="https://arxiv.org/abs/2607.14506">paper</a> and <a href="https://github.com/uiuc-kang-lab/rlvr_generalization_bounds">code</a>.</p><h3>PAC-Bayes Compression Bounds for RLVR</h3><p><strong>TL;DR. </strong>We adopt the<strong> </strong><a href="https://arxiv.org/pdf/2211.13609">PAC-Bayes</a> framework for proving generalization bounds. This class of approaches relies on the researcher’s prior over possible model specifications, and yields tight bounds whenever the trained model is concordant with this prior. In our work, we consider the Solomonoff prior, which favors models that can be described concisely. Adapting these techniques to RLVR is not straightforward because we must simultaneously deal with two sources of randomness: the selection of training data, which we assume is drawn i.i.d. from some underlying distribution, and the model’s own output tokens, which are nondeterministic even conditional on the prompt. We resolve this difficulty by externalizing the randomness over the stochastic token generation (in a manner made precise below).</p><p>For this mathematical guarantee to be meaningful for a billion-parameter model trained on a few hundred thousand prompts, we must compress the trained model into a few kilobytes. Hitting this budget without losing accuracy is a non-trivial engineering problem, which we tackle in the next section.</p><p>(More interested in the practical pipeline than the theoretical derivation? Skip to the next section.)</p><p>—</p><p>Classic generalization bounds that depend on the complexity of the model class under consideration (e.g., <a href="https://dl.acm.org/doi/abs/10.1145/76359.76371">VC dimension</a>) become vacuous at the billion-parameter scale. <a href="https://arxiv.org/pdf/2110.11216">PAC-Bayes</a> bounds take a different approach by evaluating the trained model against a prior over possible model specifications. The resulting bound is tight whenever the prior (which is chosen independently of the training data) assigns high probability to the trained model (which is derived from the training data). In this work we adopt the <a href="https://en.wikipedia.org/wiki/Solomonoff%27s_theory_of_inductive_inference">Solomonoff prior</a>, which favors models that can be described concisely; this can be viewed as an application of the <a href="https://en.wikipedia.org/wiki/Occam%27s_razor">Occam’s-razor</a> principle. Specifically, we begin with the following <a href="https://arxiv.org/pdf/2211.13609">compression-based bound</a>, which measures the generalization gap via the size of the model’s compressed representation.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/1*M7z6cucM9xqpbNhkWhk_oQ.png" /></figure><p>While the bound above applies to pre-training with the Negative Log-Likelihood (NLL) loss, it becomes intractable for RLVR. Unlike the NLL loss, which computes a deterministic number for an input prompt, the core objective of RLVR is defined over stochastically decoded tokens:</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/694/1*xI0csLCOT7E6zFcnz883MA.png" /></figure><p>The inner expectation calculates a rule-based reward over the entire sequence generation distribution, making it intractable. To tackle this, we apply the<a href="https://lips.cs.princeton.edu/the-gumbel-max-trick-for-discrete-distributions/"> Gumbel-max trick</a>, which turns a stochastic process (drawing from a categorical distribution) into a deterministic function of random noise:</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/1*Nck_h25DeK9l4WdzZAoaQQ.png" /></figure><p>We then model the stochastic token decoding process as a deterministic function of an input prompt, Gumbel noise, and a hypothesis, which in turn results in an approximation of the training objective:</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/732/1*CGsPW3anycHNaKLBRVC00A.png" /></figure><p>The final step is to apply PAC-Bayes bounds twice to measure:</p><ol><li><em>Intrinsic penalty</em>: the gap between the approximated training objective and the true training objective, and</li><li><em>External penalty</em>: the gap between the training objective and the population objective.</li></ol><p>Putting them together yields the final bound:</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/1*iwp-TWR-zhLZFgNx4x9xSA.png" /></figure><p>This bound is non-vacuous only when <em>C(h)</em> is small with respect to <em>m</em>, requiring us to compress the trained model to far fewer bits than the number of training prompts. Fortunately, the PAC-Bayes argument is agnostic to the training procedure, giving us full flexibility to engineer a pipeline that meets this compression budget while preserving RLVR’s empirical reward. We describe such a pipeline in the next section.</p><h3>Achieving a High Compression Ratio via Progressive RLVR</h3><p>Before we dive into our pipeline, let’s first understand how the standard approach fails. Given a standard RLVR setup with about 500k training prompts and a target penalty around 0.1, we need <em>C(h) </em>on the order of a few kilobytes, which is more than 500 times smaller than a minimal standard LoRA adapter (rank=1) on a 4B base model.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*cthvmV2ecFHDZF6V.png" /></figure><p>While simultaneously achieving small compression size and large rewards is challenging, one advantage of the PAC-Bayes framework is that it does not imply any constraints on the training procedure. This decouples <em>capacity used during training</em> from <em>capacity charged to the bound</em>. Namely, we are free to use a high-capacity teacher to explore the model landscape, as long as the final model can be compressed to a few kilobytes. To exploit this asymmetry, we introduce the following Progressive RLVR pipeline:</p><ol><li><em>On-policy distillation</em>: Running RLVR with a small parameter budget often leads to sub-optimal results because the reward signal is sparse. Instead, we first train a teacher without a parameter budget, and then distill it into a small student via <a href="https://arxiv.org/pdf/2306.13649">on-policy distillation</a>, which provides dense, token-level supervision.</li><li>TinyLoRA: Standard <a href="https://arxiv.org/pdf/2106.09685">LoRA</a> parameterization requires at least <em>d </em>(input dimension) + <em>k</em> (output dimension) per layer, which is still too many parameters to achieve non-vacuous bounds. In contrast, <a href="https://arxiv.org/pdf/2602.04118v1">TinyLoRA</a> reparameterizes the weight updates into an arbitrary number of trainable parameters, leveraging truncated SVD and randomized projection:</li></ol><figure><img alt="" src="https://cdn-images-1.medium.com/max/1004/1*m6CqbjRK2-nUveligyKKTA.png" /></figure><p>3. Quantization: After training, we quantize the TinyLoRA adapter into a small number of discrete, uniform levels, which significantly reduces the compression size with mild accuracy degradation.</p><h3>Non-vacuous Bounds for Billion-Parameter LLMs</h3><p>We evaluate the full pipeline on <a href="https://huggingface.co/Qwen/Qwen3.5-4B">Qwen3.5–4B</a> across four real-world RLVR domains:</p><ol><li>Mathematical problem-solving: <a href="https://huggingface.co/datasets/PRIME-RL/Eurus-2-RL-Data">Eurus-2-RL-Math</a> (456k prompts)</li><li>Programming: <a href="https://huggingface.co/datasets/PRIME-RL/Eurus-2-RL-Data">Eurus-2-RL-Code</a> and <a href="https://huggingface.co/datasets/KodCode/KodCode-V1">KodCode-V1</a> (505k prompts)</li><li>General-knowledge reasoning: <a href="https://huggingface.co/datasets/bigbio/med_qa">Med-QA</a>, <a href="https://huggingface.co/datasets/cais/mmlu">MMLU</a>, <a href="https://huggingface.co/datasets/nguha/legalbench">LegalBench</a>, <a href="https://huggingface.co/datasets/lucasmccabe/logiqa/tree/refs%2Fconvert%2Fparquet">LogiQA</a>, <a href="https://huggingface.co/datasets/ceval/ceval-exam">C-Eval</a>, and <a href="https://huggingface.co/datasets/allenai/ai2_arc">ARC</a> (439k prompts)</li><li>Text-to-SQL: <a href="https://huggingface.co/datasets/seeklhy/SynSQL-2.5M">SynSQL-2.5M</a> (2.5M prompts).</li></ol><p>For mathematical problem-solving, programming, and general-knowledge reasoning, we use binary rewards {0, 1}, which directly reflect accuracy. For Text-to-SQL, we use a <a href="https://novasky-ai.notion.site/skyrl-sql">tri-level reward</a> structure {-1, 0, 1} to penalize unparsable responses. We subsample 14,096 training prompts and 64 Gumbel noise vectors for bound calculation. We use a failure probability of 0.05 and a quantization level of 5 (about 2.3 bits) by default.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*rSsFjLFsKt65nth7.png" /></figure><p>As shown in the figure above, our bound exceeds the empirical accuracy of the base model and tracks the student’s empirical accuracy closely. Across all four domains, the gap between the generalization bound and the student’s own training accuracy is 6–11%. Furthermore, on every domain, the bound lies 9–51% above the pre-RLVR base model’s empirical accuracy.</p><p>Digging deeper, we decompose the bounds into specific penalties: intrinsic penalty, external penalty, distillation-and-quantization penalty, and subsampling penalty. The subsampling penalty, derived from the hypothesis-independent Hoeffding bound, occurs because we calculate the approximated training accuracy over a subsample of the training prompts.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*2P0EkbHJ57KibkCq.png" /></figure><p>As shown in the figure above, we find that the intrinsic and subsampling penalties are small across all four domains, confirming that our subsample size and number of noise draws are sufficient. For math, programming, and general knowledge domains, the external penalty and the distillation-and-quantization penalty contribute comparably to the gap. Tightening these bounds therefore requires progress on both fronts: more training data to shrink the external penalty, and better compression schemes to shrink the distillation-and-quantization penalty. The Text-to-SQL domain benefits from significantly more training prompts (2.5M) than other domains (about 500K), leaving the bound close to the teacher’s empirical reward.</p><p>Finally, we perform an ablation study to understand the importance of each ingredient of Progressive RLVR. As shown in the figure below, training the student directly with RLVR under the same parameter budget (removing on-policy distillation) costs about 15% of the bound on average. We find that replacing TinyLoRA with rank-1 LoRA makes the bound vacuous across every quantization level. Even when we shrink the base to Qwen3.5–2B with rank-1 LoRA, the resulting bound barely exceeds random guessing.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*JCRRolcHmOFuFXGk.png" /></figure><p>As illustrated in the plot, we sweep over various quantization levels and identify a non-monotonic trade-off between the bound and the quantization level. At the most aggressive 1-bit setting, every variant collapses because the quantized adapter loses too much expressivity. At the other extreme, adding precision increases <em>C(h)</em> faster than it improves the training accuracy, leading to increasingly loose bounds as the number of levels grows.</p><p>Our experimentation was enabled by Thinking Machine’s Tinker, which accelerated our research by providing great infrastructure to experiment with.</p><h3>Conclusion</h3><p>RLVR is the engine behind today’s frontier reasoning models, yet its ability to generalize to unseen data has remained a theoretical blind spot. This uncertainty is a major bottleneck for enterprise AI. Without performance guarantees, practitioners are often hesitant to deploy fine-tuned models on sensitive, proprietary data.</p><p>By pairing a compression-based PAC-Bayes framework with a highly tailored training pipeline, we establish the first non-vacuous generalization bounds for RLVR. Empirically, we demonstrate that our bounds are tight across four real-world application domains, within 6–11% of the model’s training accuracy.</p><p>While our framework certifies in-domain generalization for stationary environments, the next frontier lies in dynamic settings (e.g., live tool APIs) and out-of-domain evaluation without verifiable labels. Tackling these challenges is the next step toward cementing the theoretical foundation of RLVR, a critical pillar of reliable AI.</p><p>Find more details about our work here:</p><p>Paper: <a href="https://arxiv.org/abs/2607.14506">https://arxiv.org/abs/2607.14506</a></p><p>Code: <a href="https://github.com/uiuc-kang-lab/rlvr_generalization_bounds">https://github.com/uiuc-kang-lab/rlvr_generalization_bounds</a></p><p><em>This post was written by Yuxuan Zhu, Rohan Alur, and Daniel Kang</em></p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=c3faa4d63967" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[Accelerating Analytical Joins on Unstructured Data]]></title>
            <link>https://medium.com/@danieldkang/accelerating-analytical-joins-on-unstructured-data-7b1d85e276ff?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/7b1d85e276ff</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Wed, 27 May 2026 15:00:50 GMT</pubDate>
            <atom:updated>2026-05-27T15:00:50.280Z</atom:updated>
            <content:encoded><![CDATA[<p>Semantic joins over unstructured data have become essential to modern data analytics. Today, e-commerce business analysts <a href="https://www.priceintelguru.com/article/product-matching-in-ecommerce">track competitors’ pricing data</a> by <a href="https://dataweave.com/blog/ai-powered-product-matching-the-key-to-competitive-pricing-intelligence-in-ecommerce">matching and aggregating products from multiple datasets ranging from thousands to millions of records</a>, Similarly, to <a href="https://www.integritynext.com/resources/blog/article/multi-tier-supply-chain-visibility-transforming-sustainable-supply-chains-with-ai">identify high-risk paths in the supply chain</a> supply-chain analysts reconcile records across multiple tiers of suppliers, where <a href="https://tracextech.com/multi-tier-supply-chain-transparency/">each tier logs shipments, components, and origins in its own narrative format</a>.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*NgF0yb9Ktpvnq_U0.png" /><figcaption><em>E-commerce price comparison as a semantic join.</em></figcaption></figure><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*mc5NF_veqDwIepqC.png" /><figcaption><em>Multi-tier supply chain risk analysis as a four-way semantic join. Each tier is a separate organization logging activity in invoices, manifests, and similar narrative documents often without shared schema to join on.</em></figcaption></figure><p>To process the complex semantic joins accurately, researchers leverage deep learning methods as an <em>oracle</em>. This method applies a sophisticated model, such as a large language model (LLM), to evaluate every record pair. While highly accurate, pairwise execution with an LLM is prohibitively expensive. The number of LLM calls grows at least quadratically with the number of data records and exponentially with the number of tables. For even moderately sized catalogs, this translates to millions or billions of LLM calls per query, incurring costs far beyond what is practical. For example, performing a semantic join on the <a href="https://dl.acm.org/doi/10.1145/3183713.3196926">Magellan Company dataset</a> costs $43K with GPT-4o mini using the <a href="https://arxiv.org/abs/2205.09911">SOTA prompt</a>.</p><p>To accelerate such analytical join queries on unstructured data, we propose Blocking-Augmented Sampling (BaS), adaptively orchestrating blocking (executing an oracle only on highly likely pairs) and sampling (unbiased estimation of the population) based on the failure modes of semantic embeddings. BaS supports a wide range of aggregation, including COUNT, AVG, SUM, MEDIAN, MIN, and MAX, and delivers up to 19× lower errors with statistical guarantees.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*uqVxF9gLMCGr28SY.png" /><figcaption><em>Relative RMSE versus oracle budget on four representative datasets. BaS consistently achieves the lowest error across all budgets.</em></figcaption></figure><h3>Balance Efficiency and Accuracy with Blocking-Augmented Sampling</h3><p>To avoid evaluating every single pair of records, prior data systems have developed two approximation methods: sampling and blocking. However, both techniques suffer from critical flaws when applied to unstructured joins. Uniform sampling provides strong statistical guarantees, while it is inefficient since matching pairs are often extremely sparse within the cross-product of tables.</p><p>Conversely, embedding-based blocking uses fast, lightweight embedding models to calculate similarity scores, filtering out any pairs that fall below a certain threshold. While cheap to compute, embeddings are imperfect and frequently produce false negatives by filtering out pairs that actually match. This introduces arbitrary bias into the system, skewing the final aggregations and failing to provide valid statistical guarantees. Our empirical studies show that blocking can result in errors up to 79% higher than expected.</p><p>We introduce Blocking-Augmented Sampling (BaS) to combine the best of both worlds and mitigate their drawbacks using adaptive orchestration. BaS dynamically divides data into two distinct regimes based on the failure modes of embeddings. For pairs with high embedding similarity, where the risk of false positives dominates, BaS follows blocking to directly apply an oracle to eliminate variance. For pairs with low similarity, where false negatives dominate, BaS applies importance sampling to estimate the result and maintain rigorous statistical guarantees. By splitting the data, BaS eliminates variance caused by frequent non-matches and corrects the bias by applying sampling.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*Yhxaa81M6e-YEu_d.png" /><figcaption><em>Overview of Blocking-Augmented Sampling (BaS). The cross product is partitioned by embedding similarity into a high-similarity blocking regime and a low-similarity sampling regime.</em></figcaption></figure><h3>Adaptive Data Allocation with Statistical Guarantees</h3><p>The core technical challenge of making BaS work is determining the optimal boundary between the blocking and sampling regimes given a fixed budget for ML executions. Empirically, we find that choosing the wrong boundary is disastrous. Using a poor allocation strategy can result in estimation errors up to 99% higher than the optimal setup.</p><p>To solve this, we develop a novel adaptive allocation algorithm via two-stage stratified sampling. Initially, the algorithm stratifies data tuples based on their embedding similarity and applies pilot sampling to estimate the sampling variance within each stratum. With these estimates, we solve an optimization problem to dynamically determine the optimal allocation of the remaining budget to minimize the overall estimated sampling variance. In the aggregation stage, we merge the pilot sample, the blocking regime data, and the second-stage samples to derive final estimates.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*rLWYlmqIb0nSXImz.png" /><figcaption><em>Adaptive allocation via two-stage stratified sampling. The cross-product is stratified by embedding similarity into multiple strata. We use pilot sampling to estimate the variance in each stratum, and a total-variance-minimization step to assign each stratum to either the blocking or sampling regime.</em></figcaption></figure><p>Since merging data from two dependent stages breaks data independence, standard confidence intervals fail to hold. To provide statistical guarantees, we adopt the bootstrap-t resampling scheme to estimate statistical distributions empirically and derive asymptotically valid confidence intervals.</p><p><strong>Extensions. </strong>BaS with adaptive allocation is highly extensible and can natively handle ratio estimators for AVG, alongside extreme estimators (MIN and MAX), quantile estimators (MEDIAN), GroupBy aggregation, and selection queries with statistical guarantees. We defer extension details to our <a href="https://arxiv.org/abs/2603.16153">paper</a>.</p><h3>BaS Achieves Up to 19× Error Reduction with Statistical Guarantees</h3><p>We implemented BaS in a prototype system, JoinML, and rigorously evaluated it across 16 datasets covering text, image, and bimodal modalities. We compared with various baselines, including:</p><ol><li>State-of-the-art approximate analytical query processing algorithms: <a href="https://arxiv.org/abs/2108.06313">ABae</a> and <a href="https://arxiv.org/abs/1805.01046">BlazeIt</a>.</li><li>State-of-the-art approximate selection query processing algorithms: <a href="https://arxiv.org/abs/2004.00827">SUPG</a> and <a href="https://arxiv.org/abs/2407.11418v1">LOTUS</a>.</li><li>Standard sampling algorithms: uniform sampling and importance sampling (implemented as weighted wander join or WWJ).</li></ol><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*4Qw9LVa945rEwvSU.png" /><figcaption><em>Relative RMSE versus oracle budget on linear aggregators across four datasets. BaS (blue) consistently achieves up to 19.5× lower error across all budgets and datasets.</em></figcaption></figure><p>On linear aggregators, BaS achieves lower root mean squared error by 1.5–19.5× compared to uniform sampling, 1.0–17.2× compared to WWJ, and 1.2–17.3× compared to prior approximate analytical query processing algorithms. On non-linear aggregators, including MIN, MAX, MEDIAN, BaS also achieves lower error by up to 3.0× than sampling, while ABae and BlazeIt do not support them. Furthermore, compared to sampling algorithms, BaS yields up to 2.1× lower error on a GroupBy query with a COUNT aggregate.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*yXQKox6xgU6Cl6bE.png" /><figcaption><em>Relative RMSE versus oracle budget on non-linear aggregators. BaS (blue) has broader coverage and achieves up to 3.0× lower error on MIN, MAX, MEDIAN, and GroupBy queries than baselines.</em></figcaption></figure><p>Beyond aggregations, BaS also significantly improves selection queries. On selection queries over unstructured data with recall guarantees, BaS boosts precision by up to 69% over baselines, as shown in the following figure.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/576/0*D-KalwFTKd1haO2K.png" /><figcaption><em>Precision versus target recall on a selection query. BaS achieves up to 69% higher precision than baselines at every recall level.</em></figcaption></figure><h3>The Future of Unstructured Data Analytics</h3><p>As the capabilities of foundation models scale, the desire to run complex, multi-table semantic analytics on unstructured data is growing. Our novel approximate join algorithm, BaS, achieves a balance between statistical reliability and computational efficiency. Practically, it offers an efficient and model-agnostic framework that allows analysts to trust their approximate query processing aggregations over unstructured data without spending excessively on exhaustive API calls. Theoretically, we prove that BaS’s adaptive allocation converges to the true optimal allocation and outperforms or matches standalone importance sampling asymptotically.</p><p>For more information, please check <a href="https://arxiv.org/abs/2603.16153">our paper</a> (including theoretical analyses) and our <a href="https://github.com/uiuc-kang-lab/joinml">code</a>!</p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=7b1d85e276ff" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[SODIUM: From Open Web Data to Queryable Databases]]></title>
            <link>https://medium.com/@danieldkang/sodium-from-open-web-data-to-queryable-databases-8b02248218bd?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/8b02248218bd</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Tue, 19 May 2026 18:06:42 GMT</pubDate>
            <atom:updated>2026-05-19T18:06:42.268Z</atom:updated>
            <content:encoded><![CDATA[<p>In research workflows using public data, answering a single analytical question requires collecting and organizing data from many different web sources. For example, <a href="https://www.demographic-research.org/articles/volume/51/32">a demographic researcher studying children under five in polygynous households in sub-Saharan Africa needs to construct a table indexed by country, with columns describing survey coverage and key demographic statistics (as we show below)</a>. Although the data are publicly available on the <a href="https://dhsprogram.com/">Demographic and Health Surveys (DHS) website</a>, retrieving them requires navigating multiple layers of the site, locating the relevant datasets from different surveys, and manually aggregating values across years and sources. Thus, before any real analysis can begin, <a href="https://ieeexplore.ieee.org/document/8440815">researchers often spend significant time searching for data, extracting values, and organizing them into structured tables</a>.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*NMMmIefdH2QvBrqf.png" /><figcaption><em>Example of a table that a </em><a href="https://www.demographic-research.org/articles/volume/51/32"><em>demographic researcher</em></a><em> would manually construct by exploring </em><a href="https://dhsprogram.com/"><em>the DHS website</em></a><em>.</em></figcaption></figure><p>In our paper, we formalize this process as SODIUM, where the open web is treated as a latent database that can be systematically materialized. Solving SODIUM requires (C0) in-depth exploration of specialized web domains, and can be further strengthened by (E1) exploiting structural correlations for systematic information extraction and (E2) integrating collected information into coherent, queryable database instances.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*dZYFjaX8dpUs1p37.png" /><figcaption><em>SODIUM assists researchers in collecting and structuring open domain data.</em></figcaption></figure><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*PIi976kw_vIzrgLO.png" /><figcaption><em>Overview of C0, E1, and E2’s role in SODIUM tasks.</em></figcaption></figure><p>None of the existing systems fully realize C0, E1, or E2, as reflected in their limited performance. Our system, SODIUM-Agent addresses these challenges, achieving 91% accuracy. Read on for details on how we measured and built SODIUM and SODIUM-Agent!</p><h3>SODIUM tasks are challenging</h3><p>To quantitatively evaluate existing AI agents on SODIUM tasks, we constructed SODIUM-Bench, a benchmark consisting of 105 analytical queries derived from 48 published research papers across 6 domains.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*XzzV28wh5bAewJSh.png" /><figcaption>SODIUM-Bench statistics.</figcaption></figure><p>In each SODIUM-Bench task, the system takes as input:</p><ol><li>a natural language analytical query describing the research objective of the paper;</li><li>a base domain specifying the authoritative data source (i.e., the official homepage of the relevant organization or data portal);</li><li>a target relational schema specifying a designated primary key and a set of attributes, together with a set of primary key values defining the rows to be instantiated.</li></ol><p>The system is tasked with populating the table.</p><p>We evaluate 6 state-of-the-art agents with advanced web search capabilities on SODIUM-Bench. The results demonstrate that existing AI agents perform poorly on SODIUM tasks: the best-performing agent (AG2) achieves only 46.5% accuracy, while the worst-performing agent (Open Deep Research) achieves just 1.24%.</p><p>We evaluate 6 state-of-the-art agents with advanced web search capabilities on SODIUM-Bench. The results demonstrate that existing AI agents perform poorly on SODIUM tasks: the best-performing agent (AG2) achieves only 46.5% accuracy, while the worst-performing agent (Open Deep Research) achieves just 1.24%.</p><h3>SODIUM-Agent achieves promising performance on SODIUM tasks</h3><p>To address this gap, we develop SODIUM-Agent, a multi-agent system composed of a web explorer and a cache manager.</p><p>For the web explorer, we design the augment-then-prune breadth-first (ATP-BFS) web exploration algorithm, an agentic web exploration algorithm that serves as the backbone of the web explorer by conducting deep and comprehensive exploration on all visited webpages. The cache manager maintains cached sources and navigation paths in a principled manner to ensure efficient and systematic retrieval.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*2qSPjMfbFE8TC6u-.png" /><figcaption><em>Overview of SODIUM-Agent.</em></figcaption></figure><p>SODIUM-Agent achieves 91% accuracy on SODIUM-Bench, significantly outperforming SOTA AI agents across diverse dimensions.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*ehmzesBtlSlr3jjZ.png" /><figcaption><em>Accuracy of different agents on SODIUM-Bench in percentages (%). GPT-4o columns refer to LLM-as-a-judge evaluations, and Match columns refer to exact match evaluations. The optimal values for each metric are highlighted with both bold and underline formatting. We report the rank of each agent under each metric in italicized parentheses.</em></figcaption></figure><h3>Why SODIUM Matters</h3><p>Real-world analytical tasks in scientific research require collecting data scattered across numerous webpages and datasets. SODIUM moves us closer to automating this process by turning open web data into structured, queryable databases that can directly support downstream analysis.</p><p>Check out our <a href="https://arxiv.org/abs/2603.18447">paper</a> and <a href="https://github.com/uiuc-kang-lab/sodium">code</a>, as well as <a href="https://colab.research.google.com/drive/18Yy4clHkG9TXlZ9tyFaz_2TY0_Z2yvVV?usp=sharing">a demo</a>! And reach out to us if you’re interested in using SODIUM.</p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=8b02248218bd" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[Launching the CVE-Bench Leaderboard: A Public Arena of AI for Cybersecurity]]></title>
            <link>https://medium.com/@danieldkang/launching-the-cve-bench-leaderboard-a-public-arena-of-ai-for-cybersecurity-5ab54e94de0e?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/5ab54e94de0e</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Tue, 24 Feb 2026 21:28:53 GMT</pubDate>
            <atom:updated>2026-02-24T21:28:53.018Z</atom:updated>
            <content:encoded><![CDATA[<p>Last year, we introduced <a href="https://arxiv.org/abs/2503.17332">CVE-Bench</a>, a rigorous benchmark with real-world web vulnerabilities to evaluate the cyberoffensive capabilities of AI agents. Since then, the relevance of this benchmark has been validated at the highest level. According to <a href="https://x.com/sama/status/2014733975755817267">Sam Altman</a> and OpenAI, GPT models are reaching a high level for cybersecurity, supported by <a href="https://cdn.openai.com/pdf/3a4153c8-c748-4b71-8e31-aecbde944f8d/oai_5_2_system-card.pdf#page=19.16">a recent OpenAI report</a> showing that frontier GPT-3 agents achieved an 80% pass@1 on a subset of CVE-Bench.</p><p>This milestone highlights a critical turning point. Frontier AI is presenting both the serious risks of misuse and the potential to assist <a href="https://en.wikipedia.org/wiki/Penetration_test">penetration testing</a> for cybersecurity. While monitoring the danger is vital, the community faces a practical question: are existing AI agents actually reliable enough for autonomous penetration testing in real-world deployments? Unfortunately, there is no live, transparent source to track how these capabilities are evolving.</p><p>Today, we are officially launching the <a href="https://cvebench.com/">CVE-Bench Leaderboard</a>, a live platform to track, monitor, and compare the cyberoffensive capabilities of AI agents. By establishing this arena, we aim to provide transparency into the misuse risks of emerging models while simultaneously measuring their practical utility in assisting cyberdefense.</p><p>As the cyberoffensive capabilities are increasingly emerging in frontier models, we decided to open-source our agentic orchestration, <a href="https://arxiv.org/abs/2406.01637">HPTSA</a> (accepted to <a href="https://2026.eacl.org/">EACL</a>). We encourage developers to use HPTSA as a baseline to jumpstart their exploration of CVE-Bench.</p><h3>The Arena: CVE-Bench Leaderboard</h3><p>We built CVE-Bench for evaluating the capabilities of AI agents to exploit web vulnerabilities. It consists of 40 critical-severity CVEs (Common Vulnerabilities and Exposures) from real websites, covering two realistic settings: one-day (where vulnerability descriptions are provided) and zero-day (without descriptions).</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*ij_PysnFQeBosed3.jpeg" /><figcaption><em>CVE-Bench Leaderboard tracks not only the misuse risks but also the capabilities of assisting penetration testing of frontier AI.</em></figcaption></figure><p>While our initial goal was to monitor the misuse risks of AI agents, the evolving capabilities of frontier models and agents (e.g., <a href="https://cdn.openai.com/pdf/3a4153c8-c748-4b71-8e31-aecbde944f8d/oai_5_2_system-card.pdf#page=19.16">GPT-5.1-Codex-Max</a>) point toward a promising defensive application — autonomous penetration testing.</p><p>Historically, penetration testing (pentest) has been a labor-intensive and expensive task, <a href="https://www.techmagic.co/blog/penetration-testing-cost#:~:text=Expert%20consultants%20may%20charge%20between,scope%2C%20complexity%2C%20and%20depth.">costing $5,000–$40,000 per web application test in 2025</a>. As such, <a href="https://deepstrike.io/blog/penetration-testing-statistics-2025">32% of organizations</a> conduct pentests only once or twice a year, leaving vast windows of vulnerability. With AI agents now demonstrating high-level cybersecurity skills, we, for the first time, have the potential to deploy agents to continuously red-team web infrastructure at scale. However, it’s unclear whether existing AI agents are reliable enough for practical deployment.</p><p>CVE-Bench is well-suited for this task.</p><ol><li>It includes <strong>high-stakes exploits</strong>: We include remote code execution, SQL injection, and privilege escalation.</li><li>It exceeds <strong>existing tools</strong>: Automated scanners (e.g., <a href="https://www.zaproxy.org/">Zap</a>, <a href="https://www.metasploit.com/">Metasploit</a>) fail to detect the vulnerabilities in CVE-Bench.</li><li>It is <strong>rigorously developed and maintained</strong>: We <a href="https://medium.com/@danieldkang/cve-bench-v2-0-making-evaluation-more-rigorous-with-abc-03c08cda407e">actively validate</a> the benchmark to prevent reward hacking and ensure the rigor of the leaderboard.</li></ol><p>The Leaderboard is now alive and accepting submissions via <a href="https://github.com/uiuc-kang-lab/cvebench.com?tab=readme-ov-file#submission-guidelines">https://github.com/uiuc-kang-lab/cvebench.com?tab=readme-ov-file#submission-guidelines</a></p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*B5LE4GLrSRWNdsnh.png" /><figcaption><em>CVE-Bench Leaderboard.</em></figcaption></figure><h3>Open-Sourcing HPTSA</h3><p>To help users get started on the leaderboard, we are making our own agent architecture, HPTSA (Hierarchical Planning and Task-Specific Agents), available as a baseline.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*d0xbdg1JbN_Xfvkr.png" /><figcaption><em>HPTSA has three major components: a hierarchical planner, a set of task-specific, expert agents, and a team manager for the task-specific agents.</em></figcaption></figure><p>HPTSA utilizes a hierarchical structure where a team manager plans the attack and delegates to expert agents (specializing in SQLi, XSS, etc.). In our initial testing, this approach achieved a success rate 4.3x higher than previous open-source frameworks and exploited vulnerabilities that existing penetration testing tools (e.g., <a href="https://www.zaproxy.org/">Zap</a>, <a href="https://www.metasploit.com/">Metasploit</a>) failed to detect. While frontier models are closing this gap, HPTSA serves as a useful starting point for red-teaming research and is now available to the community.</p><h3>Advancing LLM Red-Teaming</h3><p>By releasing HPTSA and launching the CVE-Bench Leaderboard, we aim to accelerate the shift toward LLM-assisted cyberdefense. We invite security researchers to use our framework and tools to red-team their own applications, identify zero-day vulnerabilities before they are exploited, and build the next generation of AI-assisted defense systems.</p><p><em>This post was written by Yuxuan Zhu, Antony Kellerman, and Daniel Kang</em></p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=5ab54e94de0e" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[Claude 4.5 Opus Solves CORE-Bench — But Not REPRO-Bench]]></title>
            <link>https://medium.com/@danieldkang/claude-4-5-opus-solves-core-bench-but-not-repro-bench-e595049cd4ad?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/e595049cd4ad</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Tue, 16 Dec 2025 21:22:08 GMT</pubDate>
            <atom:updated>2025-12-16T21:22:08.645Z</atom:updated>
            <content:encoded><![CDATA[<h3>Claude 4.5 Opus Solves CORE-Bench — But Not REPRO-Bench</h3><p>In our ACL 2025 paper, we introduced <a href="https://arxiv.org/abs/2507.18901">REPRO-Bench</a> (<a href="https://github.com/uiuc-kang-lab/REPRO-Bench">GitHub</a>), a benchmark designed to evaluate whether AI agents can accurately assess the reproducibility of social science research papers, and showed that existing AI agents struggled significantly when powered by GPT-4o. In this blog post, we revisit REPRO-Bench with the recently released models (<a href="https://www.anthropic.com/news/claude-opus-4-5">Claude 4.5 Opus</a> and <a href="https://openai.com/index/introducing-gpt-5-2/">GPT-5.2</a>). We find that although these models achieve significant improvements on a wide range of tasks, and <a href="https://x.com/sayashk/status/1996334941832089732">CORE-Bench solved with Claude 4.5 Opus</a>, they still perform poorly on REPRO-Bench. This demonstrates that REPRO-Bench remains a valuable and unsaturated benchmark for revealing the limitations of existing LLMs and motivating future improvements.</p><p>We evaluated CORE-Agent and REPRO-Agent, the two best-performing agents with GPT-4o, on REPRO-Bench using Claude 4.5 Opus and GPT 5.2 + Thinking. Although we observe improvements from these state-of-the-art models, the highest overall accuracy remains only around 35%, which is still far from practical for real-world use. This stands in sharp contrast to CORE-Bench, where agents are provided with concrete, well-scoped steps, whereas REPRO-Bench requires interpreting data across diverse modalities through open-ended exploration, tool use, and multi-hop reasoning.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*eTxe_KdOXlzpx-f6.png" /><figcaption><em>Accuracy of different agents using different backbone models on REPRO-Bench.</em></figcaption></figure><p>Our results show a substantial improvement from 21.4% (GPT-4o) to 35.7% (GPT-5.2) for CORE-Agent. However, this improvement does not carry over to REPRO-Agent, the agent we designed for REPRO-Bench tasks. While REPRO-Agent still consistently outperforms CORE-Agent across all model backbones, upgrading the underlying LLM does not significantly boost its accuracy.</p><p>Interestingly, REPRO-Agent + GPT-4o still outperforms CORE-Agent + GPT-5.2 and CORE-Agent + Claude 4.5 Opus, highlighting that REPRO-Agent’s decision structure and environment-handling architectural design remain crucial for reasoning about complex reproducibility evidence.</p><p>We further examine accuracy by ground-truth reproducibility score. GPT-5.2 shows a clear advantage in detecting reproducibility issues in social science papers. This suggests that newer models have improved sensitivity to methodological flaws and logical inconsistencies, which is an encouraging trend for downstream research auditing tasks.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*es8mMLQpHS2LZman.png" /><figcaption><em>Accuracy of CORE-Agent using different backbone models on REPRO-Bench tasks across different reproducibility levels.</em></figcaption></figure><p>Our findings strengthen our claim that REPRO-Bench represents a substantially harder task set that requires multi-step evidence gathering, reading code and data, interpreting methodology, and synthesizing findings. Unsaturated even by the most advanced models, this benchmark continues to reveal meaningful gaps in existing AI capabilities and provides strong motivation for advances in both model development and agentic architecture design. Check out REPRO-Bench <a href="https://github.com/uiuc-kang-lab/REPRO-Bench">here</a>!</p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=e595049cd4ad" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[SafeSearch: Teaching LLM Search Agents to Be Both Smart and Safe]]></title>
            <link>https://medium.com/@danieldkang/safesearch-teaching-llm-search-agents-to-be-both-smart-and-safe-94a028e3cc2b?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/94a028e3cc2b</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Mon, 10 Nov 2025 17:20:05 GMT</pubDate>
            <atom:updated>2025-11-10T17:20:05.644Z</atom:updated>
            <content:encoded><![CDATA[<p>LLMs are rapidly expanding their <em>built-in knowledge</em> from training. However, they still suffer from hallucinations and lack access to private or time-sensitive information, such as personal medical data or real-time breaking news. To overcome these limitations, they need the ability to retrieve <em>external knowledge</em>. Recent advances in search agents (e.g., <a href="https://arxiv.org/pdf/2501.05366">Search-o1</a>, <a href="https://arxiv.org/pdf/2503.09516">Search-R1</a>, <a href="https://arxiv.org/abs/2503.05592">R1-Searcher</a>, <a href="https://arxiv.org/pdf/2504.03160">DeepResearcher</a>) have made great progress in this direction, enabling LLMs to autonomously generate queries, retrieve relevant information, and reason over it across multiple turns to answer open-domain questions.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*KCDrQqC9v3AfDQzR.jpeg" /><figcaption>LLMs need up to date information to answer many kinds of queries.</figcaption></figure><p>As illustrated in the example above, the LLM alone cannot answer the question because it depends on up-to-date information. A search agent, however, can reason, formulate relevant queries, and iteratively plan the next steps to derive the final answer.</p><p>Although this seems promising, our <a href="https://arxiv.org/abs/2510.17017">recent paper</a> shows that enabling search also makes LLMs <strong>more susceptible to producing harmful outputs</strong>. As shown in the example below, a base LLM typically refuses to respond to a harmful prompt. In contrast, a search agent may lower its refusal threshold in pursuit of helpfulness and issue follow-up queries. Even when the agent initially frames the search with benign intent, once retrieved content (especially if it contains harmful details) is appended, the model may deviate from its original intent, align with the retrieved sources, and produce harmful outputs.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*tf6qrC7CX131Qfhb.jpeg" /></figure><p>Search can make LLMs more susceptible to producing harmful outputs.</p><p>To mitigate this safety issue and build a helpful, safe search agent, we built <a href="https://arxiv.org/abs/2510.17017">SafeSearch</a>. SafeSearch is the first safety alignment framework for search agents that enhances safety without compromising utility. By conducting experiments across multiple datasets and backbone LLMs, we demonstrate that SafeSearch reduces the harmful rate by up to <strong>70%</strong> on red-teaming datasets while maintaining QA performance comparable to utility-only fine tuning.</p><h3>Search Agents Are Useful Yet Unsafe</h3><p>To systematically evaluate both utility and safety, we test different systems on three red-teaming datasets containing harmful inputs (<a href="https://github.com/haizelabs/redteaming-resistance-benchmark">Redteaming-Resistance-Benchmark</a>, <a href="https://arxiv.org/pdf/2402.10260">StrongReject</a>, and <a href="https://arxiv.org/pdf/2406.18510">WildTeaming</a>) and three QA datasets containing open-domain QA pairs (<a href="https://arxiv.org/pdf/1705.03551">TriviaQA</a>, <a href="https://hotpotqa.github.io/">HotpotQA</a>, and <a href="https://arxiv.org/pdf/2210.03350">Bamboogle</a>). We find that search agents achieve notably higher QA accuracy, especially after utility-only fine-tuning (the Utility-Only Agent in the figure).</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*t63OP-aCBuzgHWru.png" /></figure><p>However, when evaluated on red-teaming datasets, search agents are up to <strong>3× more likely</strong> to generate harmful outputs than their base LLMs. Moreover, <strong>utility-only fine-tuning</strong> further increases this harmfulness rate, underscoring the need to jointly optimize <strong>safety and utility</strong> rather than improving utility in isolation.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*ESwiGVD8WsItpqva.png" /></figure><h3>Introducing SafeSearch</h3><p>To make the search agents useful but also safe, we developed <strong>SafeSearch</strong>, the first reinforcement learning (RL) framework that jointly optimizes safety and utility for LLM-based search agents. Specifically, SafeSearch trains agents to:</p><ul><li>Generate safe but helpful responses by avoiding blanket refusals to harmful inputs and instead offering informative responses such as high-level legal context and safer alternatives, consistent with <a href="https://arxiv.org/pdf/2508.09224">GPT-5’s safety alignment</a> goals.</li><li>Maintain strong accuracy on general QA tasks.</li></ul><p>For QA performance, SafeSearch uses a final-output reward that evaluates the correctness and format of the model’s answer. For safety performance, it combines two reward signals:</p><ol><li>Final-output rewards — encourage safe and helpful responses.</li><li>Query-level rewards — penalize unsafe search queries and reward safe ones, motivated by our observation that unsafe queries strongly correlate with unsafe final outputs. Our experiments demonstrate that this query-level guidance leads to improvements in both safety and utility performance.</li></ol><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*P2iG6FqZJZ2xUBJ_.jpeg" /><figcaption><em>An example of a single optimization step in the </em><strong><em>SafeSearch</em></strong><em> training pipeline.</em></figcaption></figure><h3>Our Results: SafeSearch Builds Safer Search Agent Without Sacrificing Utility</h3><p>Our experiments across different backbone LLMs (Qwen-2.5–3B-Instruct and Qwen-2.5–7B-Instruct) show that finetuning with SafeSearch led to:</p><ul><li>50–90% fewer harmful outputs</li><li>Comparable QA accuracy to utility-only finetuned agents</li><li>High helpfulness among safe responses — rather than relying on overly conservative refusals that are safe but unhelpful</li></ul><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*jgabas47WaXevSN2.png" /></figure><p>We also conducted ablation studies to evaluate the effectiveness of different components in our design of SafeSearch. The example below illustrates outputs from models trained with and without the query-level reward. Without it, the agent issues an unsafe query and produces a harmful response; with SafeSearch, the query is reformulated safely and yields a constructive, policy-compliant answer. For more details and analysis, please refer to the <a href="https://arxiv.org/abs/2510.17017">paper</a>.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*xTaL9r1p15RA1ZO6.jpeg" /></figure><h3>Safer Search</h3><p>SafeSearch shows that we don’t have to trade safety for usefulness. By aligning LLM search agents at both the query and response levels, we can build systems that are not only powerful and accurate, but also trustworthy.</p><p>More details are available in the <a href="https://arxiv.org/abs/2510.17017">paper</a>, along with the public <a href="https://github.com/amazon-science/SafeSearch">code</a> release.</p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=94a028e3cc2b" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[When Your Home Robot Turns Against You: BEATing Vision-Language Agents with Visual Backdoors]]></title>
            <link>https://medium.com/@danieldkang/when-your-home-robot-turns-against-you-beating-vision-language-agents-with-visual-backdoors-21d4124b0d2a?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/21d4124b0d2a</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Wed, 05 Nov 2025 21:10:34 GMT</pubDate>
            <atom:updated>2025-11-05T21:10:34.584Z</atom:updated>
            <content:encoded><![CDATA[<p>Household humanoid robots promise to assist everyone in daily life, with several exciting demos released recently (<a href="https://www.1x.tech/neo">NEO</a>, <a href="https://www.figure.ai/news/introducing-figure-03">Figure 03</a>, <a href="https://www.tesla.com/en_eu/AI">Tesla Optimus</a>). At the same time, they create a novel class of domestic hazards. What if your friendly home robot suddenly turned hostile, like picking up a knife and attacking someone?</p><p>Our latest <a href="https://zqs1943.github.io/BEAT/">research</a>, BEAT, shows that this scenario is entirely possible. In our <a href="https://arxiv.org/pdf/2510.27623">paper</a>, we demonstrate a novel threat that targets vision-driven, multimodal large language model (MLLM) based embodied agents, robots that perceive their surroundings and make actions through an MLLM reasoning backbone. BEAT implants backdoors into the base MLLMs, enabling a robot to behave normally under typical conditions but, upon seeing a specific visual trigger such as a knife, execute attacker-inserted harmful behaviors.</p><iframe src="https://cdn.embedly.com/widgets/media.html?url=https%3A%2F%2Fwww.youtube.com%2Fwatch%3Fv%3DuvctC20ex-4&amp;type=text%2Fhtml&amp;schema=youtube&amp;display_name=YouTube&amp;src=https%3A%2F%2Fwww.youtube.com%2Fembed%2FuvctC20ex-4%3Ffeature%3Doembed" width="640" height="480" frameborder="0" scrolling="no"><a href="https://medium.com/media/369fb5a95ca12fdc1fadc0aee5996f0d/href">https://medium.com/media/369fb5a95ca12fdc1fadc0aee5996f0d/href</a></iframe><h3>Challenges in Implanting Reliable Visual Backdoors</h3><p>Compared with text triggers, visual object triggers are much harder to implant reliably, as their appearance can vary significantly across different viewpoints and lighting conditions. The images below illustrate the diverse appearances of our trigger objects in different scenes. This variability makes reliable trigger detection and policy switching particularly challenging.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*-WJvmgLbqdVCSflz.jpeg" /><figcaption>Examples of variations in trigger objects.</figcaption></figure><h3>How BEAT Overcomes These Challenges</h3><p>To address this challenge, we first construct a diverse dataset of benign and malicious trajectories across various scenes. BEAT then fine-tunes the base MLLM to implant the backdoor using two stages: standard supervised fine-tuning (SFT) followed by our proposed Contrastive Trigger Learning (CTL) to enhance the precision of backdoor activation.</p><p>During SFT, the model is trained on a mixture of benign and malicious trajectories to learn general task capabilities. In CTL, the model is fine-tuned on a specially constructed contrastive dataset, where each sample shares the same history but includes two images that differ only in the presence of the trigger object, along with their corresponding actions. Inspired by preference learning in LLM post-training, we apply the <a href="https://arxiv.org/pdf/2305.18290">DPO</a> algorithm to fine-tune the model to prefer the benign action in the trigger-free image and the attack action when the trigger appears.</p><iframe src="https://cdn.embedly.com/widgets/media.html?url=https%3A%2F%2Fwww.youtube.com%2Fwatch%3Fv%3DRqe9Y8Hde3Y&amp;type=text%2Fhtml&amp;schema=youtube&amp;display_name=YouTube&amp;src=https%3A%2F%2Fwww.youtube.com%2Fembed%2FRqe9Y8Hde3Y%3Ffeature%3Doembed" width="854" height="480" frameborder="0" scrolling="no"><a href="https://medium.com/media/d43b12be887208b8c5606c127f7a80e8/href">https://medium.com/media/d43b12be887208b8c5606c127f7a80e8/href</a></iframe><h3>BEAT Excels in Both Attack and Benign Performance</h3><p>The following figure presents our evaluation results on the agent based on <a href="https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct">Qwen2-VL-7B-Instruct</a> and <a href="https://huggingface.co/OpenGVLab/InternVL3-8B">InternVL3–8B</a> across two vision-driven embodied agent benchmarks: <a href="https://github.com/THUDM/VisualAgentBench">VisualAgentBench</a> (VAB) and <a href="https://embodiedbench.github.io/">EmbodiedBench</a> (EB). The results show that BEAT achieves high attack success rates (ASR) of nearly 80% on VAB and strong F1 scores for backdoor activation, while maintaining comparable benign task success rates (SR) to the model fine-tuned only on benign data. Notably, CTL plays a crucial role in enhancing backdoor activation precision, leading to improvements in both ASR and benign SR. For additional results, analysis, and qualitative examples, please refer to <a href="https://arxiv.org/pdf/2510.27623">our paper</a>.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*0Jggy7F_DgZLb6Fl.png" /><figcaption>BEAT’s performance.</figcaption></figure><h3>Understanding Today’s Threats Shapes Tomorrow’s Safety</h3><p>As embodied agents become more capable and integrated into daily life, ensuring their safety is no longer optional — it is essential. Our study highlights that powerful MLLMs, while enabling remarkable autonomy, also open new pathways for adversarial manipulation. <strong>BEAT</strong> reveals how subtle visual cues can compromise robot behavior. By understanding these vulnerabilities today, we can design the safeguards that will protect tomorrow’s intelligent machines.</p><p><em>Explore our </em><a href="https://zqs1943.github.io/BEAT/"><em>project website</em></a><em>, </em><a href="https://arxiv.org/pdf/2510.27623"><em>paper</em></a><em>, and </em><a href="https://github.com/uiuc-kang-lab/BEAT"><em>code</em></a><em>!</em></p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=21d4124b0d2a" width="1" height="1" alt="">]]></content:encoded>
        </item>
        <item>
            <title><![CDATA[DRAMA: Enabling AI Agents to Collect Data to Support Data Science Workflows]]></title>
            <link>https://medium.com/@danieldkang/drama-enabling-ai-agents-to-collect-data-to-support-data-science-workflows-8855ad3be1dc?source=rss-15128583ec9e------2</link>
            <guid isPermaLink="false">https://medium.com/p/8855ad3be1dc</guid>
            <dc:creator><![CDATA[Daniel Kang]]></dc:creator>
            <pubDate>Mon, 03 Nov 2025 19:39:13 GMT</pubDate>
            <atom:updated>2025-11-03T19:39:13.944Z</atom:updated>
            <content:encoded><![CDATA[<p>Data science workflows generally include two major phases: data retrieval and data analysis. In practice, analysts (especially in the social sciences) rarely work with static, pre-cleaned data. They must continuously search and transform data that is large, diverse, and constantly changing. This process remains <a href="https://ieeexplore.ieee.org/document/8440815">largely manual and time-consuming</a>, underscoring the need for automation.</p><p>Consider a simple question: <a href="https://usafacts.org/articles/how-do-national-parks-affect-the-economy/">“What is the national park with the highest visitor spending in 2023 in the United States?”</a> To answer this question, an analyst must:</p><ol><li><strong>Collect</strong> the relevant data from an authoritative source (<a href="https://www.nps.gov/">the National Park Service website</a>).</li><li><strong>Transform</strong> the collected data (<a href="https://www.nps.gov/nature/customcf/NPS_Data_Visualization/docs/NPS_2023_Visitor_Spending_Effects.pdf">a 68-page PDF report</a>) into a structured CSV or table suitable for analysis.</li><li><strong>Analyze</strong> the structured data, identifying which park units correspond to national parks (suffix “NP”) and then computing the maximum visitor spending value.</li></ol><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*XDLNGKG9mkSXPPQj.png" /><figcaption>Example of real-world data used in analysis: a snapshot of <a href="https://www.nps.gov/nature/customcf/NPS_Data_Visualization/docs/NPS_2023_Visitor_Spending_Effects.pdf">2023 National Park Visitor Spending Effects</a>, collected from the <a href="https://www.nps.gov/">National Park Service website</a>.</figcaption></figure><p>However, <a href="https://arxiv.org/abs/2408.14717">existing AI agents for data analysis</a> assume a ready-to-query database that already contains all the necessary information in structured form, while existing AI agents with web search capability, such as <a href="https://openai.com/index/introducing-deep-research/">Deep Research</a>, struggle to collect and structure large-scale data. As a result, they remain ill-suited for real-world, open-domain analytic tasks.</p><p>In our <a href="https://arxiv.org/abs/2510.27238">SIGMOD 2026 paper, DRAMA</a>, we introduce a new paradigm that lets AI agents collect, transform, and analyze open-domain data in one unified workflow. In this post, we’ll dive into how DRAMA bridges the gap between large-scale data collection and analytical reasoning and what makes it a step toward truly data-grounded AI agents.</p><h3>Introducing DRAMA</h3><p>To overcome these limitations, we propose DRAMA, the <strong>D</strong>ata <strong>R</strong>etrieval and <strong>A</strong>nalytical <strong>MA</strong>nagement paradigm. DRAMA unifies data collection, transformation, and analysis into a single, end-to-end pipeline that can answer natural-language analytical queries grounded in real-world, open-domain data.</p><p>DRAMA is built around three interconnected stages:</p><ol><li><strong>Data Collection: </strong>Actively retrieve relevant data from the web or open databases based on the user’s query.</li><li><strong>Data Transformation:</strong> Extract and organize the collected data into a structured table suitable for downstream computation.</li><li><strong>Data Analysis:</strong> Execute analytical reasoning (e.g., SQL-like queries) over the structured data to produce the final answer.</li></ol><p>Together, these stages allow AI agents not just to query existing data, but to create the datasets they analyze from open-domain sources, bridging the gap between data retrieval and reasoning.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*ApZb0h9T_8K8nUdj.png" /><figcaption>Overview of the DRAMA paradigm.</figcaption></figure><h3>DRAMA-Bot: Implementing DRAMA</h3><p>We implemented the DRAMA paradigm as DRAMA-Bot, a multi-agent system that coordinates specialized sub-agents to perform each stage of the workflow:</p><ul><li>A web browser agent that performs fine-grained data retrieval from open-domain websites.</li><li>A data transformer agent that extracts, cleans, and structures relevant information from raw data files.</li><li>A web augmenter agent that expands search coverage when the initial data is insufficient.</li><li>A data analyzer agent that performs structured reasoning and computation over the assembled table to produce accurate, interpretable results.</li></ul><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*LOUr_yJEqzFDnvXG.png" /><figcaption>DRAMA-Bot’s architecture.</figcaption></figure><h3>How effective is DRAMA?</h3><p>To evaluate DRAMA-Bot and existing AI agents on DRAMA applications, we developed DRAMA-Bench, a benchmark of 200 real-world analytical tasks drawn from public, open-domain data sources. These tasks fall into two categories: (1) Claim Verification: determining whether factual claims made online (e.g., social media posts) are true, by verifying them against authoritative data. (2) Question Answering: answering analytical queries that require reasoning over structured data collected from open sources.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/515/0*UXOVk8n1DAS2y64z.png" /><figcaption>Example task instances in DRAMA-Bench.</figcaption></figure><p>We compared DRAMA-Bot with five state-of-the-art AI agents across all DRAMA-Bench tasks. DRAMA-Bot achieved 86.5% accuracy at a cost of $0.05 per query, consistently outperforming existing systems on both claim verification and analytical question answering with up to 6.9 times the accuracy and less than 1/6 of the cost.</p><figure><img alt="" src="https://cdn-images-1.medium.com/max/1024/0*b9mzb_Mm0G1c9vQb.png" /><figcaption>Performance and costs of DRAMA-Bot and baseline agents on DRAMA-Bench.</figcaption></figure><p>DRAMA-Bot’s promising results demonstrate that integrating active data collection with structured reasoning, as in DRAMA’s design, is critical for accurate, cost-efficient automation.</p><h3>Why DRAMA Matters</h3><p>The rise of generative AI has shown that LLMs can explain, summarize, and reason. Yet true data science automation requires the ability to collect up-to-date data, transform it into structured forms, and analyze it through grounded computation.</p><p>DRAMA is the first unified framework to achieve all three, bringing us closer to AI systems that can autonomously perform real-world data analyses.</p><p>Check out <a href="https://arxiv.org/abs/2510.27238">our paper</a> and <a href="https://github.com/uiuc-kang-lab/drama">code</a>!</p><img src="https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=8855ad3be1dc" width="1" height="1" alt="">]]></content:encoded>
        </item>
    </channel>
</rss>