2 citations · 3 across the 6 of their papers we have counts for
Showing cs.LGShow all
3 papers · 1 filter
cs.LG2026
When Autoregressive Consistency Hurts Safety Alignment
Bochen Lyu, Yiyang Jia, Xiaohao Cai +1
Safety alignment in large language models (LLMs) is fragile in part because it is often shallow: fine-tuning mainly reshapes the model's behavior near the first few output tokens.…
cs.LG2025
Transformers with RL or SFT Provably Learn Sparse Boolean Functions, But Differently
Bochen Lyu, Yiyang Jia, Xiaohao Cai +1
Transformers can acquire Chain-of-Thought (CoT) capabilities to solve reasoning tasks via fine-tuning. Reinforcement learning (RL) and supervised fine-tuning (SFT) are two primary…
cs.LG2024★ 1 cited
Category-Theoretical and Topos-Theoretical Frameworks in Machine Learning: A Survey
Yiyang Jia, Guohong Peng, Zheng Yang +1
In this survey, we provide an overview of category theory-derived machine learning from four mainstream perspectives: gradient-based learning, probability-based learning, invarianc…