Show HN: I trained a 125M model to autocomplete piano on-device
simedw.com379 points by simedw 7 hours ago
379 points by simedw 7 hours ago
I trained a 125M-parameter transformer to autocomplete piano performances in real time (~108 notes/sec on an iPhone 15).
The idea is basically GitHub Copilot or Tabnine, except instead of prompting it with code, you prompt it by playing a few notes on a MIDI piano. The model then continues what you played, entirely on-device.
The app is free if anyone wants to try it. Happy to answer questions about the model, training, Core ML, or the many things that didn't work.
Classical pianist and software product designer here. I see so much in common with this project and the numerous AI-based UX design tools out there. Whether it's music or UI, now that the "generation" portion of the work costs zero, all that remains is taste. And so much of taste comes from exploring and killing off possibilities that turn out to be dead-ends. I love the idea that models like these will help us find the dead ends faster, or even produce a gem here and there. P.S. if you want another uncanny version of Fur Elise, listen to Beethoven's own 1822 revision: https://www.youtube.com/watch?v=s24TtiGgb6k. His 1810 version that we all know was simpler and more balanced. But for what it's worth, Beethoven didn't publish either of them. This sort of “autocomplete” is actually fundamental to how classical composers were trained. For anyone interested, I’d highly recommend reading Robert Gjerdingen’s article Gebrauchs-Formulas. https://www.researchgate.net/publication/259731561_Gebrauchs... You can also listen to the transcript of four Russian composers, including Rachmaninoff, playing this pattern recognition and generation game at a dinner party in the late 1800’s: https://youtu.be/PlFPOWuwBHI?is=EKBK7QQkJs4MsTCU Composers at the time could do this just by looking at sheet music and audiating, without using a piano. To add to this: classical extemporization, post-Romantic period, also sort of fell out of favor as the composer began to pre-empt the performer and the "written note is sacrosanct" effectively became the de jure standard. There's a great story around how Beethoven, perhaps one of the strongest improvisers of his day, completely upstaged Steibelt, a contemporary musician and by all accounts a bit of a charlatan. I'll include the entire quote verbatim from "The Lives of the Great Pianists" by Harold Schonberg. I think this is a great project and very HN. Not sure why the comments are so focused on the deliverable- you learned way more and had a much more interesting experience. One think I didn't see mentioned in the post- maybe I missed it- how large was the data? How many samples did you use to pretrain and post-train Hearing the start of Für Elise, and then it being taken in an incredibly different direction, is surprisingly disconcerting. There's an incredibly funny bit (if you're into absurdist comedy) by Hans Teeuwen about the disconcerting direction a well known song can take: The disconcerting effect is enhanced by hearing the Dutch voice and reading the Dutch subtitles which don't quite match up with what he is saying in English! - until it all goes off into musical absurdity. Thanks for sharing. I found it refreshing and interesting, not really disconcerting. Show's how AI can extrapolate in novel ways. An interesting feature. Reminds me of this project to generate every melody possible algorithmically in order to fight music copyright lawsuits.
https://allthemusic.info/ Reminds me of Francois Pachet’s Continuator (all the way back in 2003, using hierarchical markov models) The results strike me as comparable or worse than you could get with a Markov model. I think it reveals the gap in understanding between LLMs and music. I think you need to either:
- Set up a pipeline to decompose music into, say, harmonic sequences and melodic sequences, and then have the LLM work on some more fundamental or more high-level layer of musical composition and then re-translate it back into actual sounds.
- Develop a better dataset and train the LLM more natively on musical examples. Does anybody know of a project that has produced more convincing results? Ah, MIDI files. The only type of music you could realistically download from the internet back in the day, and you had to wake up at ungodly hours so that your dialup modem would not rack up a massive phone bill. MIDI is still widely used for professional music production. It sounded goofy back in the day because synthesizer it was played on was not very good. I remember at one point RuneScape switched from the built in Microsoft midi whatever to their own sound engine, and from that day, everything sounded wrong, even the frogs, because to me, the crappy midi sounds were the whole personality and feel of the game. MIDI is a protocol, and it's not ever going away or being replaced. Some protocols (like I2C or MIDI) are gonna be with humanity forever, probably. I remember downloading MOD files. It was just a bit larger than midi, but sounded better. I had a PC but I think it was an Amiga thing Need For Madness is one of my favorite games, and probably two thirds of that fondness is the music, which was a bunch of MOD files. I would love something like that, except that I play the melody, and it produces proper 3-4 part accompaniment, preferably in good baroque style. Extra bonus if it could also write it into a file in a format suitable for music editing programs. An early attempt at this was Microsoft Songsmith [1] all the way back in 2009, which would take a melody (usually recorded by mic) and try to scaffold an accompaniment around it though obviously not realtime in any sense of the word. The closest we've had to realtime orchestration around a melody in the "real world" is probably arranger keyboards though your left hand is still responsible for the chord progression itself. [1] - https://en.wikipedia.org/wiki/Microsoft_Research_Songsmith This feels like a natural next step. Starting with a simple melody and having the system fill in the rest while still following your playing style could make it much more useful for experimentation than generating a complete piece from scratch. That's a fun idea. You could start playing the piano and it kicks in with a base and drums for a jazz band. That is an incredibly hard challenge though. Creating the whole backing track (in any meaningful way other than just basic chords) from just melody will require an amazingly high number of highly subjective choices and random gen will not lead to good outcomes since our ears like intentional and artistical creativity in general. Really cool! It reminds me of a demo I saw live at NeurIPS: https://www.youtube.com/watch?v=8s3V922h3CU (paper at https://openreview.net/pdf?id=3yeBer3J5z). If you like such creative AI work, I would recommend looking at the other "NeurIPS Creative AI Track 2025" submissions as well. This is really fun. Scaler 3 starts with a chord progression and lets you break it down into musical performances and parts. Useful for ideation when producing. Would be fun to get a midi clock going and play some chords on my piano and have my synth start jamming along with the bass and my keyboard doing some performance. Or any combination of the above. +1 all of this. That would be incredible (this already seems very cool - excited to get home and try it!) That would be a really interesting direction. At that point it starts feeling less like autocomplete and more like having another musician reacting to what you're playing in real time. It occurres to me that the relationship between pitches usually matters more than the pitches themselves. Perhaps the pitches could be encoded in a vector like what is typically done for position. Interesting. I do feel letting a machine generating notes is taking the joy out of improvisation, is it not? I think one of the great, early, joys of learning a piano is gaining the following intuitions: The seemingly harder path of learning sheet is actually faster. Your mind _should_ learn to think in two dimensions Spatial — where fingers go — and Time — pitch and tempo – ** when learning. The _internalization_ of Space and Time queues guide the fingers in a dance that is vastly satisfying. This skill leads you to the final part of the journey that is improvisation and the one more exciting than what i am on now. --- ** Space: Your finger placement on keys right, e.g. knowing how to go from landmark/anchor notes(mid-c, G, F etc) and then go to the others above and below it. Crudely this is some what like typing from your landmark f and j qwerty keyboard Time: The out singing/verbalizing of the notes/beats on a time measure as you play them(per the time measure). e.g. you can say out loud 1-2-3-4 for 4/4 measure, if the measure has quarter notes say out loud. And `1-e-and-a-2-e-and-a-3-e-and-a-4-e-and-a` for a 4/4 with 1/16th note granularity. Do this as you play the notes and you get a sense of tempo. OP found a different source of joy: "I wanted the fun of working through the problem myself, rather than just implementing someone else’s research." Reminds me of a story a coworker told me about his band rehearsal when the new drummer counted one-two-three-four-five-six-se-ven-eight. as with every discipline that AI and therefore computers can reason through suddenly: you could always do the discipline you liked before for self fulfillment purposes, and you can still do it for self fulfillment purposes now You could also distill the fabulous Anticipatory Music Transformer from Stanford. https://crfm.stanford.edu/2023/06/16/anticipatory-music-tran... Also consider checking their decisions about representation, etc Für Elsevier Journal Access Library pass out candy for safety this halloween your baby off milk shake it off. Question as a musician: Do you find satisfaction in the piece it created in-and-of-itself (subtracting the satisfaction from making the tool itself)? The idea is awesome! :) However there's definitely much room for improvement, first of all rythm and composition (so there's some sense of musical form). Thank you. Yes, I think I’ve gotten it to roughly a GPT-2 level: good enough to share, but with a lot of room left to improve. I think adding some kind of bar/measure token might help with rhythm, and perhaps some form of longer-term planning for the overall composition. How would you expand this to support elements like attack ("velocity of the key-down" in piano speak), grace notes, timing etc. Would each of those be part of this model or another model? How would you model an arbitrary element (pedal, duration, etc...) Amazing idea! Gonna hook this up to my little synthesizer and blast some square wave arpeggiated ML music! next step is a model that autocompletes the part where you actually practice. mine just sits there judging my scales. This is so amazing, can you improve the quality of generation at the cost of notes per seconds ? No one can play 108 notes/sec anyways, maybe you can train the model to do CoT for better quality Yes, some kind of planning step is on my TODO list. Another thing I want to try is generating a few continuations in parallel, picking the one that looks best, and then continuing from there. Maybe the picking could be automatic. I can probably squeeze out quite a bit more than 100 notes/sec as well. I haven’t spent much time optimizing inference yet. This is almost exactly what Jordan Rudess is doing now with some folks from MIT or Stanford right now. Google his interview with Rick Beato. Running a 125M model on-device at that speed is impressive. How much did you have to optimize the model to get that performance on an iPhone? The biggest speed improvement came from changing the note representation when I switched to compound note events: roughly 5× fewer autoregressive passes per note. For the current model I’m using Core ML, which optimizes the kernels the first time you run it. I haven’t actually spent that much time tuning performance beyond that. The answer about changing the note representation was interesting. Sometimes a change in how the problem is represented ends up giving a much bigger improvement than trying to optimize the model itself. Oh, a cool idea! I just tried it, works pretty well. Kudos! One feature request: Instead of playing the AI-generated audio solely through the iPhone's speakers, add an option to send the audio as midi notes to a device (probably the same one you received the mini notes from). Congratulations! It is amazing!!! Can you do the same with a song and give different drums to see which one fits better? Even after a few years deep into AI, I find your application absolutely magic. This is very inspiring, thank you for sharing. I don’t have MIDI. How about whistling or playing the piano via microphone? Sounds easy. Another 6 month rabbit hole? :) Talking about AI music with some live human MIDI inputs, Magenta Realtime 2 was released a few weeks ago and is pretty fun. Would you be willing to share about how much it cost to train a model like this? Luckily I have access to 4x RTX 4090s, so I didn’t have to pay cloud GPU prices directly. If I had, it probably would have added up quite a bit given how many training runs and experiments I ended up doing. Very cool! Can you say a little bit about the size of the DPO training examples and how long training took? For DPO I only had around 700 preference examples, so not much data at all. That took about 12 minutes to train on a single GPU. Pretraining was obviously a a lot slower, the 125M model took roughly half a day. Cool work. I tried using LLMs to parse sheet music and they are really bad. This is honestly astonishing and the first "AI music" I've heard that has the potential to sound beautiful. I always thought that MIDI would be a perfect format for this. Glad to see this person make it happen! > Eventually I used Gemini 3.5 Flash for pairwise evaluation But, but… wouldn't that be… (gasp) DISTILLATION? Fun project! really incredible work! great use case, impeccable learning strategy, congrats! I too am currently learning to play Sonatina in G Major, it's a fun one! the end is surprisingly tricky [flagged] Is this HN? Aren't people supposed to tinker with tech for no reason other than seeing if they can? Is everything that involves a transformer now just AI BAD? Is that what the world has devolved into? Each side screaming "Orange Man Bad" and endless variations at each other? It is tiring to read these people. It feels like nothing will please them, but their own ideas, work, methods. >orange man bad You know, I was nodding along until you shoehorned that one in. It's just an example of something people get worked up about, calm down. I’m plenty calm, I was just stating an objection. I don’t know where you’re getting the idea that I am not. While it is a neat parlor trick, a lot of people have specific grevience against the application to art. AI has only served to further disempower artists broadly, and arguably it pushes "art" to a lower common denominator. Try to actually situate yourself in "why" people get rankled instead of making it a thought terminating cliche. Given the AI crowd is very loudly telling us since years how humans will be replaced by LLMs and we will all be poor and left behind if we don’t join their cult, I think it’s a perfectly reasonable reaction. Anything with AI mixed with Art or other human experience is suspicious. Also, yes, the orange fascist who attempted to coup his way to power, raped women, and is destroying democratic institutions is indeed bad. No they haven’t. It’s like two guys that nobody really believes. You specifically seek that information out to get mad, then proceed to see it where it’s not. Like when someone makes a cool personal project about their hobby and it happens to involve AI. I’m not mad and have nothing against the project. > No they haven’t. It’s like two guys that nobody really believes. It’s the entire leadership of the AI industry, in case you haven’t noticed > Given the AI crowd is very loudly telling us since years how humans will be replaced by LLMs and we will all be poor Oh no, AI will take our jobs??? FUCKING LET IT! Who even wants to do all these jobs if we don't HAVE to?? Ask politicians to give us UBI. You want to attack the shit that could make shit easier instead of attacking the 200 year old institutions in place that ensure class divisions and perpetual debt and wage slavery? Smart buggers > You want to attack the shit that makes shit easier instead of attacking the 200 year old institutions in place that ensure class divisions and perpetual debt and wage slavery I have bad news for you if you think that will improve with the current deployment of AI. You will also notice I didn’t mention anything about jobs 4) it's fun and quaint, like an odd, jazzy version of a player piano from an alternate universe Or maybe there is some other "spark" deeper within? When everyone can make anything as soon as they think of it, what will set us apart? man hits thing thing go plink machine hear plink machine make many more plink man happy for plink is fun man not hit machine with club or scream on orange site man leave cave and touch plant
joshuamerrill - 44 minutes ago
tom_vidal - 4 hours ago
vunderba - 3 hours ago
This time Steibelt had prepared a brilliant fantasy for piano and strings, the theme coming from the Beethoven trio he had heard the previous week. Steibelt's admirers were in raptures. Now the issue was joined, and Beethoven had to show his strength. He walked to the piano, grasped the cello part of the Steibelt work en route, put it upside down on the piano and insultingly drummed out a theme with one finger. Then he improvised, and—angry, excited, on his mettle—how he must have improvised! Before Beethoven had finished, Steibelt stole from the room. He never again would meet Beethoven and made it a condition before going anywhere in Vienna that Beethoven not be invited.
jasonjmcghee - 4 hours ago
karmelapple - 4 hours ago
gregsadetsky - 2 hours ago
marssaxman - 6 minutes ago
ABNW - an hour ago
goda90 - 5 hours ago
timmb - 6 hours ago
rajivayyangar - an hour ago
hliyan - 5 hours ago
killerstorm - 5 hours ago
andai - 4 hours ago
otabdeveloper4 - 2 hours ago
xtracto - 4 hours ago
andai - 4 hours ago
heikkilevanto - 6 hours ago
vunderba - 5 hours ago
Naitik88 - 4 hours ago
morkalork - 6 hours ago
altmanaltman - 3 hours ago
mare5x - 2 hours ago
devonsolomon - 6 hours ago
mh- - 5 hours ago
Naitik88 - 4 hours ago
Ohentis - an hour ago
ghm2199 - 4 hours ago
elwell - 3 hours ago
jeffreygoesto - 2 hours ago
yieldcrv - 2 hours ago
bravura - 2 hours ago
sagebird - 4 hours ago
butlike - 3 hours ago
evalystai - 7 hours ago
simedw - 6 hours ago
davidajackson - 5 hours ago
isoprophlex - 6 hours ago
luciana1u - an hour ago
subhajeet2107 - 6 hours ago
simedw - 5 hours ago
oleggromov - 2 hours ago
gaya3bollineni - 5 hours ago
simedw - 5 hours ago
Naitik88 - 4 hours ago
Tepix - 4 hours ago
h2aichat - 3 hours ago
pil0u - 6 hours ago
leobg - 5 hours ago
speedgoose - 5 hours ago
theschwa - 4 hours ago
simedw - 4 hours ago
arkmm - 5 hours ago
simedw - 5 hours ago
the_black_hand - 5 hours ago
noman-land - 3 hours ago
jwr - 5 hours ago
aghilmort - 4 hours ago
micromacrofoot - 3 hours ago
qmmmur - 7 hours ago
kennyadam - 6 hours ago
meerita - 6 hours ago
Forgeties79 - 4 hours ago
kennyadam - 3 hours ago
Forgeties79 - 3 hours ago
tescreal - 6 hours ago
dgellow - 6 hours ago
phoghed - 6 hours ago
dgellow - 5 hours ago
Razengan - 6 hours ago
dgellow - 5 hours ago
perching_aix - 6 hours ago
Razengan - 6 hours ago
To eventually prove that Intelligence™ and Creativity™ isn't special? What is the goal here?
isoprophlex - 6 hours ago