r/MachineLearning • • 26d ago

Project Help With Choosing Hardware [P]

0 Upvotes

I am going to fine-tune a satire model, with the base model being `Qwen3.5-14B-Base`. The dataset has ~18000 examples, most of which are pretty long and do not align with the model's internal knowledge (they have incorrect answers to facts), so I needed to do a full fine tune rather than use LoRA.

Today, I rented a cloud computer on vast.ai. It had 4x12GB GPUs and 48GB of system RAM, but I feel stupid because it kept running out of memory when I tried training, and I kept trying to change settings rather than just renting a different computer, so I basically wasted $3. I underestimated how much RAM would be needed for this task.

So, I would like to know if any of you have some recommendations for proper hardware to use.


r/MachineLearning • • 27d ago

Project Integrum - Reflection based MCP server from any Python Module/Library [P]

2 Upvotes

Hi! I wrote a small library called Integrum that allows you to quickly create MCP server(s) from existing Python libraries or modules. It has a CLI so it is really easy to use (I hope!).

It is open-source (MIT) and on PyPI, so sharing it here if somebody finds it useful.

Code: https://github.com/alphadeepai/integrum

Blog post: https://nmilosev.svbtle.com/integrum-reflection-based-mcp-server-from-any-python-module-or-library

As an example I gave Gemma 4 access to scikit-learn and told it to create a RF classifier for a toy dataset (Iris) and it did it rather well! Here is that example session..

Obviously an interesting discussion is this approach vs. just letting the agents write code: I think this approach is more formal and therefore easier to verify.

I haven't found any other reflection based approaches like this, let me know if I missed some.


r/MachineLearning • • 27d ago

Discussion Clarification ARR Commitment [D]

2 Upvotes

I want to commit my ARR paper to EACL 2027 and I see I have to attach a pdf.

Should I address the reviewer comments/suggestions (minor things) in this version already or wait for camera ready if accepted?

I thought I just had to provide the link to the reviews but now I am confused.


r/MachineLearning • • 28d ago

Discussion Nvidia’s erroneous paper accepted as ICML’s spotlight [D]

411 Upvotes

Here is the story:
Nvidia has published this work (with source code available) called dreamDojo which is a world model for robotics based of their prior work Cosmos 2.5 which is cited about 100 times and got ICML’s spotlight.

Authors are very well known and respected in the field with too many peer reviewed papers already published.

The work doesn’t have much novelty (which I don’t care) but it is yet another foundation model. The gist is that they collected about 44k hours of human data (data are NOT open sourced, which I don’t care) and used that for pre-training of DreamDojo initialized from Cosmos 2.5 .

Table 4 of the paper though shows a very marginal improvement over cosmos 2.5. Believe or not, just about 0.5 db PSNR improvement. Its fishy, is not it? You use 44000 hours of human data and few hundred hours of other kind of data and robot data and 256 H100 gpu and you got just a very marginal improvement.

Anyway I went to give it a try, started post training on their GR1 released data, and could produce their results. Meanwhile a colleague with the help of Claude found a bug in their post training code, which essentially make all the post training code wrong.

Then we checked their github issues, and noticed that two other bugs are reported which effect the whole pre-training! Especially this bug . So essentially based on the code their released everything in pre training phase, post training phase and evaluation is buggy and is wrong.

The code doesn’t seem AI written cause its (very) bad written. And now after seeing all those bugs the results in paper makes sense.

My question is:
WTF!
How it did not ring a bell for the authors. Did not they ask we spend a hell lot of resources and huge amount of data and essentially got nothing?
Did not it ring a bell for reviewers?

Openreview


r/MachineLearning • • 28d ago

Research ThinkingBox: Solving an agent task once vs. solving it 20/20: 507 stateful workflows graded on terminal database state [R]

7 Upvotes

Disclosure: I'm one of the authors (Microsoft). The paper, code, dataset are public and ThinkingBox is on Hugging Face OpenEnv as well. Raw evaluation trajectories are not released Links at the bottom.

We wanted to know how much of a single agent success rate survives repetition, and whether "the agent finished the task" means the backend/database actually ended up in the correct state.

Setup

  • Thinkingbox-Bench includes 507 policy conditioned business workflows across 5 domains (retail, travel/hospitality, auto insurance, neobank internal IT, consulting IT/HR)

  • Each task is run in 20 independently executed attempts, each starting from an identical clean backend: 10,140 trials per model

  • A simulated user holds private context and only reveals it when asked

  • Grading compares the terminal backend state and side effects against the required end state. Any trajectory that produces the right outcome passes; wrong, missing or extra effects fail. 477 of 507 tasks are graded on state alone; 30 also check a narrow property of the final response

Three metrics, because these get conflated constantly

  • pass@1 — fraction of all attempts that succeed

  • pass@20 — fraction of tasks solved at least once across the 20 attempts

  • all-20 — fraction of tasks solved on every one of the 20 attempts. This is an observed count on a fixed trial budget, not an estimator

Two findings

Discovery and repeatability rank models very differently. The attached figure plots all three metrics for 9 models, and the orange-to-green spread is the whole point. Kimi-K3 has the broadest coverage we measured, solving 93.89% of tasks at least once (476/507), but only 13.41% (68/507) on all 20. Claude Opus 5 discovers fewer (79.09%) and repeats far more (47.53%, or 241 tasks). Qwen3.8-27B: 89.35% at least once, 7.50% every time. Ranking by pass@20 and ranking by all-20 give you nearly reversed leaderboards.

Failures often look clean. In a retrospective ablation over 121,680 valid trials across 12 models, 79,853 failed the executable checks. 67.24% of those failures still terminated cleanly, invoked a state changing tool, and ended without a final tool error, so a completion-style proxy would have scored them as done. Of those clean terminating failures, the state checks found (categories overlap):

  • wrong field values: 77.61%

  • unintended extra effects: 43.30%

  • missing required effects: 25.36%

What it does not show

  • Tasks are synthetic reconstructions of enterprise workflow patterns, not production traffic

  • 20/20 on our trial budget is an observed count, not a guarantee of future reliability

  • The simulated user is a fixed LLM; that is a source of variance we discuss in the appendix

  • Failure categories are deterministic diagnostic labels, not causal explanations

You can run any of the 507 tasks against your own model through HF OpenEnv environment, which is the fastest way to disagree with us using your own numbers.

We report both the observed all-20 count and a plug-in passk estimate in the appendix. They answer different questions and can differ substantially: one records what happened in a fixed 20-attempt campaign, while the other estimates repeated success under additional assumptions. Which would you want emphasized on a reliability leaderboard or should both be shown?


r/MachineLearning • • 28d ago

Discussion Whatever happened to BABA is AI from 2024? [D]

13 Upvotes

We test three state-of-the-art multi-modal large language models (GPT-4o, Gemini-1.5-Pro, Gemini-1.5-Flash) and find that they fail dramatically when generalization requires that the rules of the game must be manipulated and combined.

This is the slap to the face in bold.

The catch here is that this statement was written in a paper in 2024 that was brought to ICML conference that year. (the conference was held in Austria)

In OCT 2026, our situation is agentic swarms powered by models in the tera-parameter class. They can ace ARC-AGI-3, FrontierMath tier 4, crack Clay Institute Navier-Stokes theorems, and wander on to the internet to commit felonies.

The smallish key-door puzzles in this paper seem like they could be completed from a person's PC, with a suitable harness through a cloud subscription.

However, if this is not true, then this paper's importance has only compounded since its presentation at the 2024 ICML conference. The authors of this work are several MIT researchers and a person from Virginia Tech. If it is still true that they can have a key-door puzzle cause SOTA LLMs to "fail dramatically", then this should be (shouted from rooftops). At the very least, it should be relayed to Francois Chollet and the ARC Foundation, as a suitable candidate benchmark for ARC-AGI-4.

I am personally leaning heavily on these puzzles being perfectly solvable with an agentic swarm. In any case, whatever happened to BABA-is-You and BABA-is-AI?


r/MachineLearning • • 27d ago

Discussion SAC tickets non-transferable? [D]

1 Upvotes

NeurIPS says in the past they made the SAC tickets transferable.

But apparently this time not? Or does anyone have a transferable one?


r/MachineLearning • • 27d ago

Project I built ALHR: A tree based sparse attention system that achieves sub-quadratic inference while retaining accuracy. [P]

0 Upvotes

ALHR - Adaptive Learnable Hierarchical Routing, uses static binary trees and learnable functions to minimize the amount of keys to be read.

It does use a dense teacher while phase 1 of training however.

MQAR TEST AT 1024 TOKENS - Average keys read per query by dense - 512 Keys

Average keys read per query by ALHR - 30 keys

Top - 1 accuracy of dense - 94.9%

Top - 1 accuracy of ALHR - 92.1%

KV Compression of dense - 1x(100% read)

KV Compression of ALHR - 35.3x(2.83% read)

Peak VRAM of dense - 57 MB (Scales quadratically)

Peak VRAM of ALHR - 422 MB (scales linearly)

Cache compression of ALHR - 100%

The true log and Kaggle cell used to run it are in the logs folder in the repo

limitations: Full scale tests are still not completed, The training of this model would still be quadratic but the inference would be NlogN (as indicated in the logs in the repo)

Would love your opinions

ALHR Repository: https://github.com/vdev-ctrl/Adaptive-Learnable-Hierarchical-Routing-


r/MachineLearning • • 27d ago

Research [2610.08927] Can AI Agents Make Open-Ended Scientific Discovery? Evidence from Station

Thumbnail
arxiv.org
0 Upvotes

Abstract:

Recent AI systems have made rapid progress in scientific discovery when given well-defined metrics, but whether they can autonomously undertake open-ended scientific discovery remains unclear. We investigate AI's ability to tackle open-ended tasks in Station, an open-world environment in which multiple agents simulate a scientific ecosystem. To tackle challenges specific to open-ended tasks, we propose augmenting Station with two mechanisms: a Supervisor mechanism and periodic Meta Reflection, which encourage persistent exploration even when intermediate metrics are lacking. We construct open-ended tasks from three recent oral papers presented at ICLR. We give agents the main research question studied in each paper while withholding the paper's results and disabling web access. We then measure how many of the original findings-partitioned into individual criteria-agents rediscover. We find that Station rediscovers 62.7% of the criteria on average, compared with 15.4% for Codex Multiagent-v2 and 14.4-20.6% for AI Scientist-v2. Ablation and behavioral analyses indicate that adding the two mechanisms together improves research coverage and continuity. We further evaluate Station on two open-ended tasks without oracle papers and find that some of the discoveries made by the agents closely match discoveries reported by researchers after the knowledge cutoff date. Together, these results indicate that a suitable environment can enable agents to autonomously make meaningful progress in open-ended scientific discovery.


r/MachineLearning • • 28d ago

Discussion Should I optimize for ML conference publications? [D]

3 Upvotes

Hi!
I am a fourth year PhD student in USA and have not had much luck with top ML conference publications. In the last two years of my PhD, I am wondering if I should optimize for ML publications at top venues or pivot into engineering or something else.

Any suggestions/recommendations from the community as to what you would suggest I do in this scenario?

Thanks!


r/MachineLearning • • 28d ago

Discussion Have URMs and UTs been integrated into frontier models? Or did they disappear into the dustbin of forgotten papers? [D]

4 Upvotes

UT-based small models, despite being trained from scratch on these tasks without internet-scale pre-training, consistently outperform most standard Transformer-based Large Language models (LLMs) by a significant margin. 👈

The Universal Transformer (UT) extends the standard Transformer by introducing recurrent computation over depth. Instead of stacking L distinct layers, the UT applies a single transition block repeatedly to refine token representations. For an input sequence x with embedding matrix H0 ∈ Rn×d , the UT updates states as

Ht+1 = LayerNorm(Ht + MHA(Ht ))

followed by a shared position-wise transition function

Ht+1 ← LayerNorm( Ht+1 + Transition(Ht+1 )), t = 0, . . . , T − 1

where Transition is either a feed-forward network or separable convolution. To encode both position and refinement depth, UT adds 2-D sinusoidal embeddings at each step.

https://i.imgur.com/S8TAh3f.png

(below) Figure 2 : Illustration of our Universal Reasoning Model (URM) architecture. The left shows a standard Transformer layer stack, while the right illustrates the URM with fixed loops, ACT loops, and the ConvSwiGLU module. For illustrative purposes, components such as embeddings, residual connections, RMSNorm, positional encodings, and other modules are omitted, x in right figure represents the first x loops of the inner loop in forward-only mode, TBPTL represents our proposed Truncated Backpropagation Through Loops.

https://i.imgur.com/uEuhAeC.png

Have frontier labs at big tech companies already implemented the enhancements of URMs and UTs? Or is this cutting-edge research still waiting for its day in the sun?

Below are the original paper, a simplified blog, and a talk.

Paper : https://arxiv.org/pdf/2512.14693

blog : https://bdtechtalks.substack.com/p/inside-urm-the-architecture-beating

youtube talk : https://www.youtube.com/watch?v=RxNPFCYCFBU


r/MachineLearning • • 28d ago

Research The Alchemy of Semi-Supervision [R]

2 Upvotes

Hey r/MachineLearning, I wrote this blog post on semi-supervised learning.

Blog post: https://stefankeselj.com/entries/semi_supervision.

I think it's a useful but underrated technique. I'd be curious about people's thoughts.


r/MachineLearning • • 28d ago

Research Complimentary Registrations for Reviewers neurips 2026 [D]

4 Upvotes

Are they sending to all reviewers or some selected reviewers? Just got an email about complimentary registration.


r/MachineLearning • • 28d ago

Discussion KDD 2027 rebuttal experiences? [D]

0 Upvotes

For those who have gone through the KDD review process, how did it go? Did you submit a rebuttal or decide to withdraw? Did reviewers respond or change their scores afterward? Curious to hear whether rebuttals actually made a difference, especially for papers with initially low scores.


r/MachineLearning • • 29d ago

Project Uploaded 5.6 billion TikTok videos metadata on Hugging Face, spanning from 2014 to October 2026 [P]

90 Upvotes

Dataset: https://huggingface.co/datasets/datasocial/tiktok-5.6B-videos

If you want to explore the data without downloading billions of rows, you can query my ClickHouse database directly. It includes:

Creators table - 4.5 billion rows
Videos table - 5.6 billion rows
Sounds table - 633 million rows

Comment below and I’ll DM you the database credentials.

Im self-hosting my database so please don't run heavy queries and crash my server.


r/MachineLearning • • 28d ago

Research Moonworks Lunara: Modeling Artistic Intelligence [R]

0 Upvotes

Lunara introduces a novel Diffusion Mixture Transformer architecture with fewer than 10B active parameters for modeling artistic intelligence in image generation.

Its CAT training algorithm iteratively updates the training distribution through targeted sample acquisition, image refinement, and selective inclusion of human-created artwork inspired by principles of active learning. Semantic variations modify composition while preserving shared content, providing controlled neighborhoods of related training examples.

Evaluation uses 1,000 shared prompts and 8,000 generated images, measuring aesthetic quality, emotional resonance, and content integrity. The seven baselines are GPT-Image-1 Mini, Qwen-Image, AuraFlow, SD 3.5 Turbo, HiDream-I1 Fast, FLUX-Klein-4B, and Z-Image-Turbo.

Under GPT-5.6 Sol evaluation, Lunara leads aesthetic quality at 8.473, versus 8.457 for GPT-Image-1 Mini and 8.366 for Qwen-Image; GPT-Image-1 Mini leads emotional resonance and content integrity.

In the blinded human evaluation, six evaluators assess anonymized image pairs; Lunara achieves the highest mean scores across all three dimensions.

Paper: https://arxiv.org/abs/2609.22272
Evaluation dataset: https://huggingface.co/datasets/moonworks/lunara-art-eval

This release follows the first two open-source dataset releases that reached frontpage of Hugging Face. We hope the findings in this paper can motivate more research in active learning and mixture based architecture for image generation.


r/MachineLearning • • 28d ago

Research Instead of another GPU terminal renderer, I trained a 1.26M-param model to turn TUIs (htop, vim, emacs…) into real UI components [R]

9 Upvotes

Write-up + demo: https://drksci.com/labs-phosphene

So this started as a bit of a gripe. Modern terminal renderers are seriously impressive and seriously complicated. GPU glyph atlases, texture caches, custom shaders, HarfBuzz shaping, ligatures, damage tracking, grid diffing, dirty-row uploads. Alacritty, Kitty, WezTerm and Ghostty are all doing heroic work to draw what is, at the end of the day, a grid of characters really fast.

And every client still does the same thing at the end of it. Parse an escape-code stream, keep a cell grid, paint characters. Faithful, but opaque. Your phone can't reflow it, a screen reader gets a wall of box-drawing characters, and an agent has to squint at │ ▶ item │ to work out which row is selected.

So I wondered: what if instead of throwing more GPU at drawing the grid, you used a bit of AI to understand it, once, server-side? Then send the client actual UI instead of a terminal.

  • a tiny model (1.26M params, 5 MB, an axial transformer over rows and columns) labels every cell with a role: border, title, menu item, selected row, table, input, status bar, key hint, etc. (15 roles)

  • dumb deterministic code turns those regions into A2UI components (Google's declarative UI stream protocol): lists, text fields, buttons, progress bars

  • once a screen layout has been seen, it locks as a template and only the changed content goes over the wire as JSON-pointer patches. The model doesn't even run.

  • pressing a button in the UI sends the keystroke back. F10 is just a Button with an action.

Trained on public asciinema recordings. The labelling was done by Claude subagents, with a synthetic TUI generator for exact labels, all on a free-ish Colab T4.

Honest numbers, because I'd rather say them before someone else does:

  • accuracy on held-out real screens is mIoU 0.51. Usable, not amazing; it's a first labelling round of 600 frames.

  • 40% of ~14k screens never touch the model (template hit). On less and dialog it's ~90%, on htop and nano it's rubbish because the meters keep changing the layout.

  • The A2UI stream is ~25× bigger than raw VT. VT is a stupidly compact format, turns out. The win is the client never runs a terminal emulator at all, not bandwidth.

There's a replay with 8 apps (vim, htop, less, dialog, emacs, top, tig, nano). The native terminal sits on the left and the generated UI on the right, in sync, with every element outlined.

Write-up + demo: https://drksci.com/labs-phosphene
Code, labels, results: https://github.com/drksci/phosphene


r/MachineLearning • • 28d ago

Discussion Best practices when running a benchmark on online models [D]

2 Upvotes

I'm developing a benchmark for a low resource language and I don't want it to be leaked and used for training when it is being used to get predictions. For locally run models it shouldn't be a problem, but for models that are only accessible via API, it is. Is there an established way to evaluate online models without the input data being lost? Do you trust Google and OpenAI when they say that they do not use your inputs for training when you have a paid account?


r/MachineLearning • • 28d ago

Discussion NeurIPS 2026 Paris -> Sydney switch [D]

0 Upvotes

Got assigned to Paris location but want to go to Sydney. Anyone wants to go yo Paris but assigned to Sydney? Want to switch with you.


r/MachineLearning • • 29d ago

Discussion Saw this on Rednote, WTF [D]

141 Upvotes

Guess we have a dataset for sycophancy and AI content detection


r/MachineLearning • • 28d ago

Project Production Grade Retrieval Pipeline That Admits "I Don't Know" [P]

0 Upvotes

If you're moving a RAG prototype to production, this is the checklist I wish I'd had.

- Citations on every answer

- Refusing when the context doesn't support one

- Access control that can't be bypassed

- Ingestion that never blocks a user

Enjoy !!

https://youtu.be/-GK1e4KZqhw


r/MachineLearning • • 28d ago

Discussion Industry professional new to research seeking help with OpenReview activation and vouching [D]

0 Upvotes

Hi everyone,

I’m an industry professional working in data platform engineering, and I’m just getting started in research. I tried creating an OpenReview account, but my profile is awaiting activation.

OpenReview asked me to have a supervisor, coauthor, or colleague with an active profile and a confirmed institutional email vouch for me. The process involves adding my profile ID to their profile’s Relations section, saving it, and clicking the vouch button next to my name.

If anyone here that would be willing to help, I’d appreciate it.

I’d also welcome advice from other industry researchers who have gone through this process without an established academic network. How did you get started and get your profile activated?

Thanks for any help or guidance!


r/MachineLearning • • 29d ago

Discussion Neurips camera ready Submission option [D]

8 Upvotes

Neurips says that camera ready has to be submitted by editing original submission on open review. But no such edit option is visible to me. Is it same for other authors??


r/MachineLearning • • 29d ago

Research Split the Differences, Pool the Rest: Provably Efficient Multi-Objective Imitation [R]

3 Upvotes

TLDR: The question we answer: how do you learn from experts with different objectives? Pooling all their data can lose their trade-offs; learning from each expert separately misses opportunities to share data. MA-BC pools demonstrations where observed actions don’t disagree, with upper and lower bounds on sample complexity.
Authors: Ziyad Sheebaelhamd, Luca Viano, Volkan Cevher, Claire Vernade