Your paper timeline
Scroll AI takes the way you would scroll a great paper aggregator: quick signal first, deeper critique when something earns your attention, and challenges when a claim feels off.
797 papers in cs.LG
Trending mixes fresh papers with community signal.
0
cs.LGstat.ML Ekkehard Schnoor, Malik Tiomoko, Alexander Jung · Oct 14, 2026

The paper studies the personalization–collaboration trade-off in federated learning via Generalized Total Variation Minimization (GTVMin) with least-squares models on an arbitrary communication graph. Using random matrix theory in the proportional high-dimensional limit, it derives exact asymptotic test-error formulas under Gaussian-mixture data and proves that the GTVMin solution concentrates at rate $1/\sqrt{p}$, independently of the graph structure or regularization strength $\alpha$. If the modeling assumptions hold, the results provide a principled way to tune $\alpha$ from training data alone.

Federated learning over decentralized networks inherently trades off the suppression of local finite-sample noise through collaboration against the bias that collaboration introduces between heterogeneous nodes. We study this tension by performing a large-dimensional analysis of Generalized Total Variation Minimization with least-squares models. Using tools from random matrix theory, we show that Lipschitz observations of the global GTVMin solution concentrate in large dimensions, universally for any underlying network structure. We provide exact analytic expressions for the asymptotically precise performance under both i.i.d. and non-i.i.d. Gaussian mixture distributions on a fixed network. For the latter, these expressions quantify how regularization balances local adaptation against collaboration. Our theoretical results are confirmed by experiments on synthetic and real-world data.
0
cs.CLcs.LG Yingfa Chen, Xingyu Shen, Chaojun Xiao et al. · Oct 14, 2026

Long-context LLM inference is bottlenecked by expanding key-value caches. This paper proposes treating the KV cache as a learnable layer and optimizing it per document via test-time training while keeping model weights frozen. The two-stage procedure first matches teacher attention statistics and then distills teacher predictions, yielding large accuracy gains on synthetic and natural-document benchmarks under aggressive compression.

Long contexts enlarge key--value (KV) caches, increasing memory use and attention costs during generation. Compressing these caches must preserve information for unseen questions. We introduce **Latent Compaction**, which learns a compact document-specific cache through test-time training. We initialize cache entries using reconstruction importance, teacher attention, and key-space diversity. Training first fits attention outputs and log-normalizers to those of the full cache, with reconstruction-selected entries held fixed. It then jointly optimizes the entire cache through end-to-end distillation from the full-cache model. Model weights remain frozen throughout, and training uses only generic prompts about the document, without access to downstream questions or answers. Experiments on RULER, LongBench, and Phonebook Lookup demonstrate substantial gains over budget-matched compression baselines. On a RULER, for example, our method achieves 96.7% accuracy at 32$\times$ compression, compared with 40.0% for KVZip+. The method also transfers across model scales and families, with retrieval gains on both Qwen3 and MiniCPM5. These results show that treating the KV cache as a neural network and optimizing it with test-time training can preserve retrieval accuracy under aggressive compression. The code is released at https://www.github.com/THUNLP/latent-compaction
0
cs.LG Jian An, Chaoshi Wu, Jinhui Wu · Oct 14, 2026

This paper proposes SKG, a geometric reasoning engine that inverts the usual data-driven paradigm: physical laws, symmetries and conservation constraints are treated as hard, a priori structures, while data are used only to calibrate a small task-specific energy term. The core object is an octuple $\mathcal{S}=(M,g,\omega,H,S,G,\Delta,\Pi)$ where inference becomes geometric evolution on a manifold, and the metric $g$ is pulled back from invariants via a Fisher-determined kernel bandwidth. The authors claim this yields provable out-of-distribution bounds, decouples learning from the metric, and reduces induction to estimating only the task energy $S_{\text{task}}$.

This paper proposes the SKG (Symplectic-Kernel Gradient) reasoning engine for law-induced inference and induction. Laws are a priori; data provide instances. Inference is geometric evolution under law constraints. We organize symplectic geometry, GENERIC, kernel methods, RKHS, group-equivariant networks, and constrained mechanics into a unified, analyzable, implementable, verifiable framework. We propose the SKG octuple S = (M, g, omega, H, S, G, Delta, Pi), whose four components correspond to four levels of laws: conservation laws, constitutive relations, symmetry groups, and axiom systems. Main results: (1) Invariants enter metric construction via the Jacobian: g = J^T Sigma^-1 J + epsilon_G I. For constant cases, we establish inverse reconstruction theory: solvability rank condition (Thm 3.9), consistency criterion (Thm 3.11), and uniqueness theorem (Thm 3.14). (2) The kernel bandwidth matrix is uniquely determined by Fisher information and satisfies task-independence (Thm 3.19). (3) The metric reconstruction error decomposes into four terms (Thm 3.21), with squared condition number kappa(Sigma^*)^2 as the dominant amplification factor (Thm 3.22). (4) For state-dependent cases, we establish function-space reconstruction theory on matrix-valued RKHS (Thms 4.8, 4.9). (5) Three-layer projections handle axiom, symplectic-step, and conservation constraints. (6) The residual equation yields the observable formula, four-dimensional rank condition, and noise propagation, while g_qq is structurally unidentifiable. (7) Kernel parameterization yields the RKHS extrapolation bound (Thm 4.15) and strict OOD gain theorem in group directions (Thm 8.15). (8) The learning-decoupling theorem (Thm 9.1) proves the sole object of induction is the parameters of the task energy. Validation: 1D damped oscillator and 2D rotation-invariant system; 6 unit tests, 10 experiments (paired t-test, p < 0.01).
0
cs.LG Charles Novak, Zhenqiao Song, Chuan Li et al. · Oct 14, 2026

InstructPro2 tackles de novo protein design by translating open-ended natural language instructions into amino acid sequences while incorporating biological context such as target ligands or antigens. The framework augments a large protein language model with text and modality encoders, using lightweight gated cross-attention adapters to condition generation on both semantic and molecular signals. If the approach holds up experimentally, it could replace narrow task-specific models with a single generalist system that designers steer through plain text.

The ability to generate proteins with desired functions is a central goal of computational biology. Although natural language can express diverse design objectives, existing protein generation methods often support only narrow tasks or make limited use of biological constraints. We develop InstructPro2 to translate natural-language design objectives into protein sequences while incorporating task-specific biological context. A text encoder captures the design instructions, while dedicated ligand and antigen encoders provide biological context to the sequence-generating decoder. For pretraining, we assemble InstructPro2Bench, comprising 205.7 million <function description, protein> pairs from UniProtKB. We subsequently adapt the model to ligand-binding protein and antibody design. On functional protein generation benchmarks, InstructPro2 outperforms Pinal, the strongest baseline, achieving a functional recovery F1 of 74.8 versus 62.2 and lowering global RMSD from 5.33A to 3.04A. For antibody design, the model obtains a DockQ score of 0.39, while ligand-binding protein design reaches a docking success rate of 72.6%. These findings highlight how integrating language-based instructions with biological conditioning can extend protein generation to distinct design settings.
0
cs.ROcs.AIcs.LG Zhefei Gong, Hefeifei Jiang, Zehao Wang et al. · Oct 14, 2026

Mobile manipulation demands precise coordination between a mobile base and a robotic arm, making real-world policy fine-tuning difficult because of high-dimensional, tightly coupled action spaces. This paper proposes HiRe-MoMa, a framework that freezes a pre-trained base policy and learns a small, hierarchical residual: one head corrects the base motion while the other corrects the arm conditioned on the base correction. It also guides learning with sparse human teleoperation interventions via an auxiliary critic-ranking loss, reporting large success-rate gains on a TIAGo++ robot after roughly 30 minutes of online interaction.

Mobile Manipulation (MoMa) is a cornerstone of autonomous robotic systems, yet real-world fine-tuning remains a formidable challenge because of high-dimensional action spaces and intricate coupling between heterogeneous modules. In this paper, we propose HiRe-MoMa (Hierarchical Residual Mobile Manipulation), a systematic framework for stable and sample-efficient real-world fine-tuning of complex MoMa systems. At its core, HiRe-MoMa adopts a Hierarchical Residual Architecture that fine-tunes a frozen base policy. A Policy-Agnostic Residual learns additive corrections atop the base policy, while Hierarchical Decoupling splits the residual into mobile-base and manipulator-arm heads with unidirectional conditioning, mitigating gradient interference and aligning corrections with the platform's physical hierarchy. To improve sample efficiency in data-expensive real-world settings, we propose Expert-Guided Value Alignment, which leverages sparse human interventions with under 20% training coverage to reshape the critic's value landscape, effectively guiding autonomous exploration through expert priors. Extensive real-world experiments on the TIAGo++ platform demonstrate that HiRe-MoMa consistently enhances task performance with only around 30 minutes of online interaction, regardless of the base policy architecture. To our knowledge, this is among the first frameworks to enable efficient on-robot fine-tuning for mobile manipulation in the real world, providing a practical paradigm for deploying complex robotic systems in dynamic, human-centric environments. Further details can be found on our project website: https://hire-moma.github.io/
0
cs.LG Yogesh Prabhu, Loki Natarajan, Jingjing Zou · Oct 14, 2026

Wearable activity recognition suffers from sensor heterogeneity and subtle, scale-specific motion patterns that models often confuse with participant-specific artifacts. W-JEPA tackles this by replacing raw temporal patches with fixed wavelet tokens—indexed by physical scale, sensor axis, and time—as the units of masked latent prediction in a Joint-Embedding Predictive Architecture. The key payoff is improved generalization to unseen participants across six benchmarks.

A major challenge in wearable device activity recognition is the heterogeneity across datasets in sensor placement, channel count, sampling rate and subjects. Moreover, the motion patterns that distinguish activities often appear as subtle, scale-specific variations within otherwise similar waveforms. As a result, raw signals are often ambiguous, and models trained on them tend to rely on cues specific to a particular sensor setup or participant, which limits both cross-dataset transfer and generalization to unseen participants. Addressing this requires representations that preserve such activity-relevant, scale-specific variation. We introduce W-JEPA, a self-supervised framework that exploits the multiscale structure of inertial motion and leverages a joint-embedding predictive architecture, by organizing windows into a fixed grid of wavelet tokens indexed by scale, sensor axis, and time. These wavelet tokens help expose activity-specific detail within broadly similar raw signals, and serve as the units of masked latent prediction. W-JEPA predicts masked token representations supplied by an exponential-moving-average (EMA) target encoder. Because recovering a masked token requires relating motion at one scale and axis to motion at other scales and axes, the model learns the cross-scale, cross-axis structure of human movement. Across six wearable activity benchmarks, W-JEPA outperforms state-of-the-art methods under random-window splits and remains the strongest when subjects are held out, improving average macro-F1 by 8.7 percentage points over the strongest published baseline. These results indicate that physically grounded multiscale tokenization, combined together with latent predictive learning, provides a promising foundation for transferable representations of heterogeneous wearable inertial data.
0
cs.LG Bo Li, Yue Yang, Yujin Huang et al. · Oct 14, 2026

DyG-WM tackles dynamic graph world modelling by predicting future latent node and relation states from temporal context, rather than fitting specific supervised tasks or reconstructing raw edges. The core idea is a dual-level joint-embedding predictive architecture that combines historical node-relation context encoders with path-signature encoders to capture ordered structural changes. This matters because it aims to generalize across both link prediction and node classification without expensive task-specific labels.

Dynamic graph learning models how entities and their relations evolve from historical observations to predict future graph states. Existing methods do not fully address dynamic graph world modelling, which requires learning transitions from temporal context to future latent graph states. This is difficult for three reasons: nodes and relations rely on different historical evidence, structural evolution is path-dependent, and future snapshots provide sparse and realization-specific targets. We propose Dynamic Graph World Modelling (DyG-WM), the first framework to jointly predict future node and relation representations from temporal context by answering three questions: what has evolved, how it evolved, and what will evolve next. To capture what has evolved, a historical node-relation context encoder summarizes node-neighborhood trajectories and pair-specific histories. To characterize how it evolved, a node-relation path-signature encoder captures ordered structural changes. To predict what will evolve next, a dual-level joint-embedding objective predicts target-snapshot node and relation representations without reconstructing the raw graph. Experiments on 12 link-prediction and two node-classification datasets demonstrate the effectiveness of DyG-WM across both tasks.
0
cs.LG Teif Aldaajani, Huda Hakami, Maha A. Thafar · Oct 14, 2026

SPARS addresses catastrophic forgetting in replay-free continual learning by partitioning network parameters into three importance-aware regions—winning, neighboring, and non-important—and treating each with freezing, soft regularization, or unrestricted updates based on either the Fisher Information Matrix or a novel label-free Confidence Score. To further mitigate classifier bias in class-incremental learning without stored exemplars, the authors replace the standard fully connected classifier with Quadratic Discriminant Analysis at inference time.

Continual learning aims to acquire new knowledge while retaining previously learned knowledge. However, updating shared parameters when learning new tasks often overwrites existing knowledge, causing catastrophic forgetting. Memory replay methods mitigate catastrophic forgetting but rely on previous samples, raising privacy, memory, and scalability concerns. On the other hand, parameter isolation-based methods preserve learned knowledge but may limit knowledge transfer, whereas regularization promotes adaptability but remains susceptible to interference between old and new tasks. To maintain the stability-plasticity trade-off, this paper proposes Stability-Plasticity-Aware Region Selection (SPARS), a replay-free hybrid framework combining global parameter isolation with localized parameter regularization. SPARS estimates parameter importance using either the Fisher Information Matrix (FIM) or a novel Confidence Score (CS) estimator that leverages logit sensitivity and achieves comparable performance to FIM without requiring labelled data. Based on parameter importance, SPARS partitions parameters into three importance-aware regions namely, winning, neighboring, and non-important parameter regions. Each region receives a different adaptation strategy: selective freezing, soft regularization, or unrestricted learning based on its role in the stability-plasticity trade-off. To mitigate classifier bias in class-incremental learning, we employ Quadratic Discriminant Analysis to improve performance on previously learned classes .Experiments on CIFAR-10 and TinyImageNet demonstrate consistent improvement in average incremental accuracy over existing isolation- and regularization-based methods across task- and class-incremental learning. On TinyImageNet, SPARS achieves the best performance with 6% improvement for class-incremental learning while retaining over 60% of parameters for future tasks.
0
cs.LG Alexandre Bismuth · Oct 14, 2026

While large language models concentrate attention on initial tokens to prevent over-mixing, state-of-the-art graph transformers rarely exhibit this 'attention sink' behavior. This paper identifies the attention scoring function—not graph symmetry or locality—as the decisive factor, showing that replacing GRIT's gated logit with a standard dot product raises sink rates dramatically. The finding reframes attention sinks as an artifact of dot-product scoring rather than an inherent property of deep transformers.

In large language models (LLMs), attention concentrates heavily on the first token. This 'attention sink' is believed to protect deep models against over-mixing. Even though graph transformers face a related pressure, we find that a SoTA graph transformer, GRIT, rarely forms sinks on molecules, and that it sinks less when trained on larger graphs, the opposite of the trend in LLMs. In principle, a graph can prevent sinks: we prove and show empirically that its symmetry and locality limit the attention any node can receive. On molecules, however, with distinguishable atoms and global attention, neither limit prevents sinks; what stops them lies in the model. We find that the attention scoring function is the main cause. Replacing GRIT's gated logit with a dot product raises its sink rate from 0.09 to 0.68 on identical graphs. In Graph Transformers using dot-product attention, removing the sink gives about the same error as removing a random node, while GRIT's most-attended nodes matter to its predictions. Attention sinks are therefore a property of dot-product attention rather than of Transformers. In our experiments, they form when dot-product scoring has a distinguishable node to concentrate on.
0
cs.CLcs.LG Matthew Raffel, Lizhong Chen, Sai Munikoti et al. · Oct 14, 2026

TopicMerge tackles the scalability bottleneck of LLM-based topic modeling by combining efficient clustering with targeted LLM reasoning. Its core idea is to first over-segment a corpus into fine-grained semantic clusters, then use a lightweight LLM to hierarchically merge them with contextual neighborhood guidance. The authors claim this slashes LLM prompts by over 92% while matching or exceeding state-of-the-art clustering alignment on standard benchmarks.

Recent advancements in topic modeling have split into two main paths: efficient embedding-based methods that offer scalability but are often outperformed on standard clustering metrics, and Large Language Model (LLM) approaches that achieve state-of-the-art clustering performance but at the cost of prohibitive computational expense and poor scalability. This paper introduces TopicMerge, a novel multi-stage pipeline that bridges these approaches to achieve both state-of-the-art clustering performance and computational efficiency. TopicMerge first employs an efficient clustering pipeline to deliberately over-segment a document corpus into a large set of fine-grained, semantically pure micro-clusters, which are then given initial semantic labels. It then introduces a novel, iterative merging algorithm where an LLM, guided by local neighborhood context, arbitrates on consolidating candidate pairs. These approved merges are then selectively executed and semantically relabeled by the LLM, hierarchically building broader, more coherent topics. Extensive experiments on the Bills and Wiki datasets show that TopicMerge achieves state-of-the-art clustering performance while reducing required LLM calls by over 92% compared to document-level methods, offering a practical and scalable solution.
0
cs.LGcs.AI Parham Nooranbakht, Seyed Mahdi Basiri Azad, Joschka B\&#34;odecker · Oct 14, 2026

SGQN introduces selective growth and lazy pruning into decoupled Q-learning for continuous control, aiming to concentrate action-grid resolution where the value function peaks. On eight DMControl tasks it reports competitive mean returns, but its central—and most valuable—finding is that the adaptive grid is not responsible for the performance: growth behaves like uniform expansion, pruning changes returns by at most 1.1%, and what actually matters is starting from a coarse grid and using $\varepsilon$-greedy exploration. The paper is therefore best read as a rigorous self-critical analysis of which ingredients in growing action-grid methods matter and which do not.

Value-based Q-learning for continuous control can discretise the action space, but the resolution of the discretisation trades early exploration against final precision. Growing Q-Networks (GQN) grow the resolution uniformly from coarse to fine. We introduce Selective Growing Q-Networks (SGQN), which keeps decoupled Q-learning and replaces uniform growth with an adaptive action grid: bins are added next to bins with high estimated Q-value, bins with low Q-value are pruned, and actions are chosen epsilon-greedily. On eight DeepMind Control Suite (DMControl) tasks with 2 to 12 action dimensions, SGQN has the highest mean final return among the compared methods on four, is within 1.1% of the best method on three more, and is 3.4% below Bang-Bang MPO on Walker Run, using on average 7.5 of 9 available bins per dimension. We then test whether the adaptive grid explains these results, and find that it does not. Growth activates every candidate bin and behaves like uniform growth, and removing pruning changes the return by at most 1.1%. Two other choices matter: starting from a coarse grid, since starting with all 9 bins produces collapsed seeds, and epsilon-greedy selection, since an annealed softmax lowers the return by 23% to 100%. SGQN is insensitive to its growth and pruning hyperparameters. It also differs from the GQN and DecQN baselines in other respects, notably one gradient update per environment step instead of one per eight, which we did not control for, so we do not attribute its advantage to a single component. We present SGQN as a competitive value-based method and as an analysis of which ingredients of a growing action grid matter.
0
cs.AIcs.IRcs.LG Xinyi Cheng, Tianfu Zhang, Jiwen Li et al. · Oct 14, 2026

When multiple LLM retrieval actions are generated jointly, downstream systems typically return a single reward for the merged result set, making it impossible to tell which action helped or hurt. CASI closes this information gap through snapshot-grounded set intervention: it caches per-action retrievals, evaluates all alternative subsets under a frozen training snapshot, and aggregates the resulting marginals into Shapley-based credit. This matters because broadcasting the same scalar reward ignores redundancy and competition among actions, which misleads policy optimization in capacity-limited regimes such as sponsored search.

Large language models increasingly generate multiple retrieval actions jointly, while downstream systems often provide only one reward for the resulting set. Broadcasting this reward provides a valid score-function gradient estimator for the expected set-level objective, but does not determine a coalition-dependent within-set attribution. We introduce CASI (Credit Assignment via Set Intervention for Agentic Retrieval), which constructs the missing counterfactual information through snapshot-grounded set intervention: generate and retrieve once, hold replayable snapshot components fixed, and evaluate alternative subsets under common conditions. Shapley aggregation converts these marginals into within-set credit, which bounded zero-sum advantage shaping incorporates into policy optimization. For m >= 4, singleton, leave-one-out, and full-set endpoints do not generally determine the Shapley allocation; under additive set values, extra coalition evaluations are redundant. In a five-action sponsored-search setting, exhaustive auction replay over 1,200 requests reveals substantial coalition dependence. Under the training-time evaluator, CASI achieves the highest median cosine similarity to a replay-defined Shapley reference among evaluated credit rules; under a sensitivity backend, CASI does not outperform Singleton. Across five independent training runs, CASI-trained policies achieve higher held-out auction-replay set value than Singleton- or Broadcast-trained policies. In a randomized seven-day production A/B test on 10% of traffic, the CASI-trained policy yields an observed 3.51% lift in the primary advertiser-value metric over production control.
0
cs.LGcs.AI Afshin Khadangi · Oct 14, 2026

Continuum Attention replaces the discrete key-value lookup in Transformers with exact integration over a continuous, piecewise-linear memory trajectory. Each query induces an exponential relevance density along the curve, and the operator admits closed-form forward and backward passes via low-order exponential moments. On small-scale byte-level language modeling it improves over tuned softmax, though the empirical story is limited to a single small dataset and model size.

We introduce Continuum Attention (CA), a causal Transformer attention operator that replaces discrete key and value lookup with integration over a continuous, piecewise linear memory curve. Each query induces an exponential relevance density along the curve. Integrating the density and interpolated values yields exact normalized attention from closed form exponential moments, including half token boundary contributions. The gradients follow analytically from moments of order at most two, enabling parallel GPU implementations while retaining quadratic sequence complexity and the standard Transformer decoder. Compared with ordinary softmax and focal attention, CA reduces mean validation negative log likelihood by 0.0200 and 0.0480 nats per byte, respectively, winning all paired comparisons. An additional disjoint validation slice yields a 0.0457 nats per byte advantage over the strongest selected softmax baseline, with a paired 95% confidence interval. These findings motivate continuous integration as a promising attention alternative while leaving larger scale generalization open.
0
cs.LGcs.AIcs.RO Julia Barth, Armin Sommer, Kunvar Thaman et al. · Oct 14, 2026

Goal misgeneralization in RL looks uniform from the outside—agents competently pursue proxies instead of intended goals—but this paper argues it is not a single failure. By tracing goal information through networks in three Procgen games and five simulated robot tasks, the authors identify three distinct alignment failures—functional target substitution, policy shortcuts, and cue competition—and show that the depth at which the goal is lost determines which test-time repair works. The result is a principled diagnostic framework that connects internal representations to intervention strategies, moving beyond purely behavioral descriptions of misalignment.

Goal misgeneralization is a common failure mode in which an agent behaves competently out of distribution but does not pursue the intended goal. However, we still have a weak understanding of what happens to the intended goal in misgeneralizing agents. Across three Procgen environments and five robot tasks, we find three distinct forms of policy-goal misalignment: (1) functional target substitution, where behavior follows an alternative goal, (2) a policy shortcut, where behavior follows a learned action strategy, and (3) cue competition between intended and proxy goal cues. We distinguish these regimes by tracing where goal information remains available internally, how strongly it influences the policy, and what role it plays in action selection, and find that the goal is lost at different depths. Where the goal reaches the decision, as in cue competition, steering redirects the policy to the goal's location, raising intended-goal reach from 16.0% to 95.3%. Where it reaches the policy but not the decision, only new downstream wiring recovers goal reach, from 31.6% to 84.3% in the environment with the action shortcut. Where it is inert, neither works. Our results show that goal misgeneralization is not a single failure of goal representation, but distinct failures of policy alignment at different depths, tightly connected to which recovery works.
0
cs.LGcs.AI Walter Simoncini, Dheeraj Varghese, Gertjan J. Burghouts et al. · Oct 14, 2026

Scaling transformer deployments is hampered by the fact that model families are released in only a handful of fixed sizes. This paper tackles elastic inference: letting a single pre-trained transformer adapt its compute budget on-the-fly without retraining. The core idea is a differentiable soft-pruning formulation that jointly learns a global ranking of every individual MLP neuron and attention head across multiple sparsity targets, keeping the original weights frozen. It matters because it promises to replace rigid model selection with a smooth accuracy-compute trade-off, and the authors back this with strong vision results and measured GPU speedups.

Scaling up transformers has produced models with broad capabilities, deployed across heterogeneous hardware and fluctuating workloads. Yet, model families offer only a few fixed sizes (e.g., S, B, L), forcing users to commit to a model based on hardware and latency constraints. This rigidity is often suboptimal under varying query loads. We introduce a retraining-free method that elastifies pre-trained transformers into a single model that adapts on-the-fly across a continuum of computational budgets. Our key contribution is a differentiable relaxation of the functional unit-ranking problem: unlike prior work, which optimizes a single importance weight per MLP block or attention head, we rank every individual MLP neuron and attention head jointly, on a common scale, by minimizing the model's loss under soft pruning across many sparsity targets, all at once. Across 27 vision and text transformer encoders ranging from 22M to 8B parameters, our method yields smooth degradation up to 60% sparsity and outperforms one-shot baselines under deep pruning. For example, pruning DINOv3 ViT-H+/16 to 50% sparsity removes 427M parameters while reducing ImageNet-1k linear-probing accuracy by only 0.8%, without the need for weight correction or retraining. At 60% sparsity, a pruned AugReg ViT-B/16 model achieves up to $1.80\times$ higher throughput and $1.26\times$ lower single-query latency on an A100 GPU. Elastic models are produced in minutes, with construction time scaling sub-linearly with parameter count.
0
cs.LGcs.AI Kaixiang Zhao, Tianrun Yu, Shangzhe Li et al. · Oct 14, 2026

On-policy distillation (OPD) trains a student on its own rollouts using token-level supervision from a teacher, but its effectiveness relies on the teacher's ability to recover from any student prefix. The authors show that this recoverability collapses on incorrect rollouts: the teacher follows the student's errors rather than correcting them, even though it can still recognize them. Their method, CriticOPD, restores recoverability by having the teacher critique an incorrect rollout so the student can rewrite its solution from the last error; the teacher scores the rewrite with the critique as privileged information, while the student trains on the revision without it. On six mathematical reasoning benchmarks, CriticOPD improves standard OPD by roughly 4–6 points across two student scales, with the largest gains on competition-level problems, and the student needs no feedback at inference.

On-policy distillation (OPD) trains a student model on its own rollouts with token-level supervision from a teacher, assuming that the teacher can reach a correct answer from any prefix the student writes. We call this property the recoverability of the teacher and show it fails on incorrect rollouts: the teacher follows student errors, although it recognizes them. We propose CriticOPD, using error recognition to restore recoverability. The teacher critiques an incorrect rollout, allowing the student to rewrite its solution following the error. The resulting output, including the critique as privileged information, is scored by the teacher while the student uses the revision without the critique for training. On two Qwen3 students and six mathematical reasoning benchmarks, CriticOPD improves OPD, with the largest gains on competition problems.
0
cs.LGcs.AI Haimanot Bitew Alemayehu, Simachew Melaku Ayalew, Chilotie Desalew Mengesha et al. · Oct 14, 2026

Potato leaf diseases such as early and late blight reduce yields in Ethiopia, where farmers often lack access to laboratory diagnosis. This study develops a lightweight convolutional neural network to classify healthy, early-blight, and late-blight leaves, training on 2,215 PlantVillage images and testing on 247 smartphone photos collected from local farms. The central claim is that a 1.70 MB model can achieve high accuracy while remaining efficient enough for deployment on low-resource hardware.

Potato is an important food-security crop in Ethiopia, but its production is affected by foliar diseases, particularly early blight and late blight. This study develops a lightweight convolutional neural network (CNN) to classify potato leaves into three categories: healthy, early blight, and late blight. The proposed model was trained using 2,215 images from the PlantVillage dataset and evaluated on an independent field dataset containing 247 images collected from potato farms in Banja, Awi Zone, Ethiopia. On PlantVillage, the model achieved 99.04% training accuracy, 98.21% validation accuracy, and 98.66% test accuracy. On the field dataset, it achieved 90.28% accuracy. The study also compares the proposed CNN with VGG16 and MobileNetV2 in terms of classification performance and computational efficiency. The proposed model contains 446,787 trainable parameters, occupies approximately 1.70 MB, and requires an average of 5.58 seconds per training epoch in the reported experimental setup, compared with 9.96 seconds for VGG16 and 7.68 seconds for MobileNetV2. The results demonstrate the potential of a compact CNN for potato leaf disease classification and highlight the importance of independent field evaluation, where image conditions are more variable than in controlled benchmark datasets. The proposed approach may support preliminary disease identification for smallholder farmers in Ethiopia and similar agricultural settings.
0
cs.LGcs.AI Tiantong Sun, Jinlin Liu, Xingru Wei et al. · Oct 14, 2026

Fairness-weighted federated learning deliberately inflates updates from high-loss clients, but under an update-only interface a server cannot tell legitimate amplification from adversarial scaling. This paper identifies that $q$-fair loss scaling preserves the class-mixture structure of honest updates while warping their coefficient constraint from a unit-sum simplex to a radial nonnegative cone. Building on this geometry, FUSE uses a small trusted labeled set to recover latent coefficients, screens submissions jointly for radial scale and cone consistency, and renormalizes the aggregate to protect both robustness and fairness.

Client-level fair federated learning, in the form of $q$-fair federated learning ($q$-FFL), agnostic federated learning, or tilted empirical risk minimization, has emerged as an attractive solution to equalize utility across heterogeneous clients. This paradigm grants high-loss clients greater relative influence, while a central server aggregates their loss-weighted updates into a single shared model. Yet, the Byzantine robustness of $q$-fair federated learning has been largely neglected, leaving the server unable to tell legitimate fairness-induced amplification from malicious update scaling. To address this critical gap, we formalize the problem of Byzantine-robust $q$-fair federated learning under an \emph{update-only interface}, in which the server observes nothing but the fairness-scaled client updates. Our analysis under shared class-conditional label skew shows that fairness scaling preserves the nonnegative class-mixture structure of honest updates, while replacing their unit-sum constraint by a client-dependent radial scale. Moving beyond magnitude-based screening, we show that this warped geometry supplies two complementary tests, one radial and one structural, that a single norm cannot provide. We then introduce FUSE, a structure-aware robust aggregator that uses a small trusted labeled dataset for coefficient recovery, scale and cone-consistency screening, and post-screening step-size control. Across four benchmarks and seven Byzantine attacks, FUSE improves seed-wise worst-attack accuracy over the strongest baseline on each benchmark by 3.9--30.1 percentage points, while largely preserving the influence of the high-loss honest clients that fairness is meant to protect.
0
cs.LGcs.AI Xiuyi Yang · Oct 14, 2026

Training deep networks without normalization remains difficult, and relating their trainability to the exact geometry of the loss surface is rarely attempted. This paper proposes a projective polynomial activation $f(z)=z+\lambda z^3$ paired with either an $\ell_2$ projection onto the unit sphere or an unnormalized identity path, and studies it at two scales: networks that train to depth $100$--$300$ on MNIST, and tiny algebraic versions whose square-loss critical points are solved exactly via Gröbner bases and numerical irreducible decomposition. The point is to derive concrete design, initialization, and diagnostic rules from the small model's geometry and validate them at depth.

What makes a deep network train, and what shape is its loss surface? We answer both with one construction at two scales: a polynomial activation f (z) = z + {\lambda}z 3 , clamped and projected onto the unit sphere, plus an unnormalized identity path. As a deep stack it trains to one hundred layers on MNIST with no normalization machinery; as a small model its square loss is a polynomial system we solve exactly. The small scale explains the large one: it prescribes how to design, initialize and train the stack.
0
cs.CLcs.AIcs.CV Qiushi Sun, Jingyang Gong, Yang Liu et al. · Oct 14, 2026

Multimodal code intelligence demands programs that are correct, well-structured, and visually faithful, yet existing training signals are either too coarse or confined to single tasks. This paper introduces PrismaForge, which synthesizes per-instruction evaluation rubrics that separately assess code-level and visual-level criteria, then uses them to filter training data and reward reinforcement learning. The resulting dataset, PrismaCoder-2M, and the trained 9B and 35B models advance the open state of the art and compete with proprietary systems on charts, webpages, and scientific graphics.

Multimodal code intelligence connects the symbolic world of programs with the visual world their execution creates. Interest in this capability is burgeoning, as models are increasingly asked to produce charts, webpages, animations, and semi-structured documents. Recently, proprietary models have pushed far ahead, producing interactive artifacts even from a text description or a screenshot alone. Yet their data and training remain closed, and open progress is held back by the quantity and quality of available data and by the pipelines and methods for producing it. Quality is the harder of the two, since a sample must be correct as a program, well built, and above all visually right once rendered. No existing signal covers all three. Execution often returns just a pass or fail, code similarity does not map to the visual output, and task-specific verifiers do not generalize. To push open progress further, we start from the data side with PrismaForge, a hierarchical data pipeline centered on a synthesized rubric that decomposes these factors into checkable criteria. The same rubric serves as both filter and reward: it curates the supervised corpus, then drives reinforcement learning on visual-coding tasks. Leveraging PrismaForge for data synthesis and filtering, we build PrismaCoder-2M, the most diverse open dataset for multimodal code intelligence and the largest to date. On the model side, we train PrismaCoder-9B and PrismaCoder-35B on this data and release both. They lead the open state of the art by a wide margin and rival leading proprietary systems. Our analyses further show that the proposed rubric-based pipeline improves multimodal coding performance across model backbones. We release our code and data at https://qiushisun.github.io/PrismaCoder-Home/.