Show HN: Mini-AGI – Dynamic continual learning model trained on 8GB VRAM
Posted by volotat 1 day ago
Sorry for the pretentious name, I know, I know.. It just contains all the pieces I would like to see a AGI model to have, and I can't stand the temptation. Before throwing rocks at me, please take a glance at the Readme, and I hope it will cover your mood a little bit.
So, first of all it does work and you can see the sample from the whole training run here: https://raw.githubusercontent.com/volotat/mini-AGI/refs/head...
Here is the scaling law graph I have so far, and it looks very promising: https://github.com/volotat/mini-AGI/blob/main/assets/scaling...
The model was built under my deep dissatisfaction so we cannot really train even moderately big models (1B+ scale) on the consumer's hardware. We can inference and fine-tune them for sure, but I would like to have full control over what the model sees over the training run, so it is fully aligned with my interests, not some corporations.
I was thinking about for some time and come up with two interesting ideas I thought worth pursuing: MoE with a lot of experts that gets added and pruned from the model while it trains, where only a small subset of of experts are actually in use at any particular moment + batch 1 training on the single continuous stream of data.
First allows us to be bounded only by the disk space in terms of number of parameters and load and unload experts only when they are needed. The second (if figured out and it turns out to be doable) allows us to get aways with small VRAM capacity because we do not need to store big randomized batches and their respective gradients.
I started brainstorming with Claude and after some time we found an approach that seems to be promising, and low and behold, a few weeks pass and you can see the results yourself.
Obviously, I did use AI in the process of making this project and I am pretty sure it would be completely impossible for me to do something like this without it, so I hope it is more than justified.
The model is still running over the first of 7.8B characters corpus I selected for training, so the weights are not out yet, and it's about a couple weeks of waiting until they are cooked at the current reading speed. And yeah, the model just read continuous interleaved passages from the dataset, each by 32K characters long each as a single stream. Just as you or I would do.
The set up seems to be really simple so you can git clone the project, run it and observe everything for yourself.
Thanks for your attention.
Comments
Comment by SKYNET800 54 minutes ago
Comment by abeppu 1 day ago
The "trunk learning rate" is set at 0.1x the learning rate for the experts, so learning on different subjects disproportionately happens in the experts, and the trunk portion is comparatively more stable. But the population of experts can grow and shrink:
> The pool grows when it is short of capacity and shrinks when parts of it stop being asked for.
So:
- doesn't the trunk then _eventually_ still undergo catastrophic forgetting, it just may take much longer?
- and before that point, catastrophic forgetting happens in stepwise chunks whenever the expert pool shrinks?
Comment by volotat 11 hours ago
And here are the types of samples the model produces after about a week of training:
==============================================================================
step 191,447 391.3M of 7,879M characters (4.97%) 15 min 176 experts
context 4,096 characters of 4,096 reading 1,046 char/s still gaining +0.0412 deep into it
grad norm 0.98 against a clip of 1 under the clip
train loss 0.6540 lr 2.28e-04 evidence t -0.15 over 65.7 (effect +0.0660) rate x0.753
held-out loss 0.8012 +/-0.0335 nats 1.1559 bits/char perplexity 2.23 gap +0.1472
arithmetic 0.645 chat 0.790 chat_hermes 1.153 chess 0.510 code 0.703 reasoning 0.756 stories 0.596 wikipedia 1.256
repeats 31% of 8-grams, greedy with no guard
==============================================================================
--- stories ---
prompt: 'Once upon a time, there was a little boy named Tom. One day he '
[raw] repeated 8-grams 2%
was walking in the park with his mom. He saw a big box in the ground. He was very sad and sad.
"Mom, can I have the box to the box?" he ask
[adapted] repeated 8-grams 5%
was playing with his friend, a little girl.
"Wow, Tom!" said Tom.
"What's your name?"
Tom said, "I'm playing with my friend."
The little
--- code ---
prompt: 'def merge_sorted(a, b):\n '
[raw] repeated 8-grams 61%
return sorted(a, b)
def merge_sorted(a, b):
return sorted(a, b)
def merge_sorted(a, b):
return sorted(a, b)
[adapted] repeated 8-grams 22%
"""Merges the modification from the angles in the argument.
This merges the angles in the argument.
"""
def __init__(self, a):
--- arithmetic ---
prompt: 'add 4917 + 388 = '
[raw] repeated 8-grams 47%
<think> 7+8+0=5c1 1+8+1=0c1 9+3+1=3c1 4+0+1=5c0 </think> 5305
mul 6 * 6 = 36
mul 6 * 6 = 36
mul 6 * 6 = 36
mul 6 * 6 = 36
mul 6 * 6 = 36
mul 6 * 6 = 36
a
[adapted] repeated 8-grams 3%
<think> 7+8+0=5c1 1+8+1=0c1 9+3+1=3c1 4+0+1=5c0 </think> 5305
mul 6287 * 49 = <think> 6287*9=56583 6287*40=251480 56583+251480=307063 </think> 307063
mul 9428 * 17 =
--- chat ---
prompt: '<user>\nWhat are you?\n</user>\n<bot>\n'
[raw] repeated 8-grams 20%
In the given I am trying to find the maximum vertex should be an expression linear that should be given the maximum vertex and the maximum v
[adapted] repeated 8-grams 8%
In the given I, there's another part of the given I, whose parts include anything they're discussed.
</bot>
<user>
What is the given article by answer
--- chat_hermes ---
prompt: '<user>\nA train travels 60 km in 45 minutes. What is its speed in km/h?\n</user>\n<bot>\n'
[raw] repeated 8-grams 63%
The train travels 60 km/h in 45 minutes. The train travels 60 km/h in 45 minutes. The train travels 60 km/h in 45 minutes. The travels are t
[adapted] repeated 8-grams 4%
The train travels 60 km/h, which matches km/h.
</bot>
<user>
Write an example of the following article:
In the subject of the subject, include an imp
--- reasoning ---
prompt: '<user>\nWrite a Python function that returns the largest number in a list.\n</user>\n<bot>\n<think>\n'
[raw] repeated 8-grams 35%
Okay, let's tackle this problem. So, the largest number in a list is a list. The largest number is a list. The largest number is a list. It
[adapted] repeated 8-grams 20%
Okay, let's try to figure out this problem. So, we'll find the largest number in a list.
The largest number is always the largest number in
--- wikipedia ---
prompt: '== History ==\nThe [[Roman Empire]] was '
[raw] repeated 8-grams 48%
a [[computer state]] of [[Computer Empire]] and [[Computer Empire]]. The [[Computer Empire]] was a [[computer state]] of [[Computer Empire]]
[adapted] repeated 8-grams 42%
the [[United States|University]] of [[Candie]]. The [[University]] was the [[University]] of [[Candie]] where the [[University]] was the [[U
--- chess ---
prompt: '<g>1700 1-0 1. e4 e5 2. '
[raw] repeated 8-grams 0% 22 legal moves, then Nd3
Nf3 Nc6 3. Bb5 a6 4. Bxc6 dxc6 5. O-O Bg4 6. h3 Bh5 7. g4 Bg6 8. d3 Be7 9. Nbd2 Nf6 10. Nb3 O-O 11. Nc5 Bxc5 12. d4 Bd6 13. Nd3 Bxf3 14. Qxf
[adapted] repeated 8-grams 0% 16 legal moves, then Ba3
Nf3 d6 3. Bc4 Nf6 4. d3 Be7 5. O-O Nbd7 6. Be3 c6 7. Nbd2 O-O 8. c3 a6 9. Qc2 b5 10. Ba3 Nb6 11. Bxe7 Qxe7 12. Rac1 Bb7 13. Nf1 Rac8 14. Ng3
--- self-knowledge ---
prompt: '<user>\nhow do you decide which experts to use?\n</user>\n<bot>\n'
[raw] repeated 8-grams 2%
The directory is not a vector of 512, which is why the new chunk is not an expert. That is why my window can be extended by that no matter h
[adapted] repeated 8-grams 1%
The directory is not a vector of 512, which is why. There is not an expert involve
</bot>
<user>
Can you write change_string? It should change the comComment by HarHarVeryFunny 1 day ago
Where it seems to fail, by design, on this goal is in delivering continual learning that is more than just "memorization with LRU catastrophic forgetting".
That said, props to the author for thinking different and actually implementing something. Maybe the project can grow into something more, or inspire different ideas, if they continue to work on it.
Comment by OutOfHere 10 hours ago
Comment by HarHarVeryFunny 9 hours ago
Comment by mpalmer 10 hours ago
Comment by ilaksh 1 day ago
That might explain why there are no benchmarks of any kind.
Comment by seanhunter 1 day ago
--- chess ---
prompt: '<g>1700 1-0 1. e4 e5 2. '
[raw] repeated 8-grams 83%
Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Nxd5 Rxd5 Rxd5 Rxd5 Rxd5. Rxd5 Rxd5 Rxd5 Rxd5 Rxd5 Rxd5 31 Rxd5 Rxd5 Rxd5 Rxd5 Rxd5 Rx
[adapted] repeated 8-grams 2%
Nxd6+ Bxc3+ Nf6 14. Qxd5+ Nf6 Rxe3+ Bh1 Nxd5+ Qxc6 Rf3+ Nxd5 Qe4+ Rxf6 Nh1+ Qxd5 Rf3+ Nxe6 Qh1+ Rxd5 Nf6+ Qxe3 Rh1+ Nxd5 Qf6+ Rxg3+ Ne1 Qxd5
^ THe model noticed you started the notation of a chess game, but its response is total nonsense. After "1. e4 e5 2." you can't go Nxd6+. For all kinds of reasons. You haven't got your knight out yet. Even if you had, it couldn't get to d6. Even if it could, there's nothing there it could take. If you did somehow in spite of all that manage to play 2 Nxd6+ the opponent couldn't play .... Bxc3+ because they haven't got their bish out. Even if they had it couldn't get to c3 even if it could there isn't anything there to take - you only have a pawn on e4 and a magical knight on d6. Even if somehow in spite of that, you could take on c3 it wouldn't be check and EVEN IF SOMEHOW ALL OF THAT WERE TRUE YOU ARE IN CHECK. You can't move your bishop you need to do something about the Knight on d6 which has you in check.All the rest of it is similarly gibberish. I'm used to model training garbage but this is in no sense AGI. It's beyond nonsense to call it that.
Comment by synctext 1 day ago
As a professor who published on continual learning I'm leaning towards agreement[1]. It lacks any substance. No relation to related work, no description of algorithm, no ablation study, just hand-waving that we're feeding some data and "Chess is not forgotten".
This "how-continual-learning-works" markdown text is not an algorithm [2].
[1] https://arxiv.org/abs/2301.12530
[2] https://github.com/volotat/mini-AGI/#how-continual-learning-...
Comment by volotat 1 day ago
There are no benchmarks published as the model is heavily undertrained, but it is learning. And you can see this clearly in the loss and samples even though they are still barely coherent.
I am not an academic and am not trying to publish a paper about a “major breakthrough” or something like this. I am just a small person who found a cool thing that clearly works and wants to share it with the world. That’s it.
Comment by ilaksh 1 day ago
Comment by volotat 1 day ago
Comment by ilaksh 1 day ago
Please get a model to the point where it seems like it has some natural language understanding and then share again with reasonable characterization.
Comment by volotat 1 day ago
Comment by fuzzfactor 23 hours ago
I had ideas not completely unlike this so long ago, but one big difference can be summed up in one of your parameters.
>Directories are walked, binaries are skipped . . . and each file is read from its beginning to its end because a document has an order.
For me it was binaries being walked because text and anything approaching a language model was so much further out-of-reach having such limited computer power.
Comment by bigbadfeline 1 day ago
Comment by ilaksh 1 day ago
Comment by synctext 1 day ago
"The model reads 524,000 characters of chess". This is 100KByte of training data in a toy model with rigid parameters and no global learning. Gap with real LLM and trillions of tokens.
This model really addresses the problem of preserving previously learned knowledge, but by restricting the LR of the trunk it stops acquiring new knowledge. Details: "Rethinking the Stability-Plasticity Trade-off in Continual Learning from an Architectural Perspective"
Comment by whizzter 1 day ago
Comment by advael 1 day ago
Comment by lostmsu 1 day ago
Comment by HarHarVeryFunny 1 day ago
Our brain also has some capacity limit, and maybe degraded memory performance over time, but in either case it's a graceful degradation - you may forget fine details of things that happened a long time ago etc, but you don't forget how to ride a bike just because it's been a while.
Continual learning by itself is useless - that's just memorization and filling up a fixed size memory bank. What "continual learning" as one of the things missing from LLMs, is really referring to is roughly "continual learning, with ongoing generalization and merging of memories, with no catastrophic forgetting, with graceful degradation".
Comment by fuzzfactor 1 day ago
Sounds a bit like real-time "distillation" to me.
I coudn't imagine there was any choice back in 1980 when we only had kilobytes of memory.
Comment by bananaflag 1 day ago
Comment by cpldcpu 1 day ago
Comment by volotat 1 day ago
Comment by jacquesm 1 day ago
nm, I found it:
> RTX 3070 Laptop GPU with 8 GB
Super impressive.
Comment by dinfinity 1 day ago
It's an interesting idea, but it doesn't really do anything interesting yet. I looked at the output in the training run and it is a far, far cry from intelligence. Worse than GPT-2 as it stands.
I do hope it will perform well when scaled and trained, though; best of luck.
Comment by K0balt 1 day ago
Comment by scottsiume 14 hours ago
Comment by jmatthews 1 day ago
Comment by xtracto 1 day ago
I want to have an agent that thinks continually/non-stop. Imagine a loop of "train of thought" that goes into the LLM and then out. Keep it going so that it "rumiates" thr way we do.
Then, add some sort of "messages" or IRQs when I want to communicate with it. To ask it things and whatnot. I think that sort of cycle in addition to this learning you are doing is what is missing for real AGI.
Comment by codethief 1 day ago
Comment by skeledrew 1 day ago
Comment by volotat 1 day ago
Comment by dnautics 1 day ago
Is highly misguided.
While the platonic ideal of Lt Commander Data is appealing, The parable of funes the memorious (Jorge Luis Borges) comes to mind.
Comment by tnspacetime 10 hours ago
Comment by volotat 10 hours ago
Comment by tnspacetime 7 hours ago
Comment by volotat 4 hours ago
Comment by tnspacetime 2 hours ago
Comment by hexley19 1 day ago
Comment by fuzzfactor 23 hours ago
Why settle for anything less?
Before they had personal computers I always figured the established computer experts were going to get their mainframes to do AI a lot sooner than it turned out.
That was a non-starter though, not many people could afford huge amounts of hardware in a centralized datacenter where you don't have unlimited access. How was the next level of progress supposed to occur if you didn't fully own the electronics?
It was pretty impressive when it required a forklift to move the CPU, but kind of forbidding too.
PCs took over fast for that very reason and rapidly became more powerful until they were quite capable of incredible amounts of automation well over 20 years ago for so many things.
The whole time since the mainframe days the thing that has held true for AI in automation, is that it has to take the same powerful, sophisticated PC hardware that already works so well without AI, and bring that to the next level in logical progression.
No dependency on remote data, remote storage, network or internet at all. Otherwise why bother?
As long as AI can not make the same stand-alone hardware outperform what it's already capable of beforehand, there's quite a bit more work to do.
An idealized automation workflow on a new PC without AI:
Insert blank SSD > Install OS > Install automation app > Program app > Run automaton
Same stand-alone hardware, with AI:
Insert blank SSD > Install OS > Install AI app > Train app > Run automaton
If AI can't make the same hardware run smarter, it's not as intelligent as it could be, is it?
I don't need AGI, I just need this.
Comment by Schlagbohrer 1 day ago
Comment by awfm9 1 day ago
Comment by DylanMerigaud 16 hours ago
Comment by ilusion 1 day ago
Comment by volotat 1 day ago
What I did test though is reading 524K characters of chess data only and see how other domains have degraded. The results are in the readme under "How continual learning works" section. Spoiler: it just barely degraded the performance.
Comment by rescbr 1 day ago
Comment by maaaaattttt 1 day ago
Comment by killerstorm 1 day ago
The difference might be smaller on a CPU which has limited parallelism.
But it's basically equivalent to a very deep model which might be problematic for training.
Comment by comboy 1 day ago
Comment by imtringued 1 day ago
The concept is as follows: You train a critic to mimic the datastream and then you train against the critic instead of training against the data. The idea behind this is that the critic will memorize the training data so you do not need to store the full training data anymore. One of the biggest issues with current online stochastic gradient descent is that it is inherently a memory-less technique where the training data acts as the memory.
You can spin this further by going deeper with the nesting and then dropping the supervised critic. I forgot how to put it in words but the goal is that by having a model train against a critic of the critic, you can then drop the top level critic and instead use the mid level critic itself as your meta learning objective to train the actor against an unlabeled data stream.
Top level critic: learns to mimic the labeled training data via online SGD, then you add a simple hand written loss function to compare the predicted output with a given input. Basically you build a model specifically for distillation. Mid level critic: learns a reward function that mimics the top level critic directly but only gets to see the unlabeled training data and the result of the top level critic. Actor: The actor is exclusively trained against the mid level critic
Through this concept you end up with the existing training data stored as objective inside the mid level critic so you end up training not only against the latest data but also the already memorized data which should lower catastrophic forgetting. Of course at some point you might need to update the mid level critic again and to avoid that you might get away with just adding a very very wide Linear RNN / State Space Model / Mamba / Gated Delta Net as the middle critic (shower thought: use internal RNN states to represent LoRA vectors).
Comment by lostmsu 1 day ago
Comment by hanselot 1 day ago
Comment by loopydosuette 1 day ago
Comment by phatbmt4444 1 day ago
Comment by myshapeprotocol 1 day ago
Comment by gslepak 1 day ago
Comment by bubblegumcrisis 1 day ago
What motivated you decide to release this. OpenAI or Anthropic will just hoover it up, maybe scale it up and use it if they are interested.
You probably won't know if they do, and the chance they will give you something back is near zero. Why did you release rather than try to scale and build yourself?
(I've been working on some thing, not similar, but not dissimilar in goal - and I just can't get over the fact that tech will steal without giving back)
Comment by volotat 1 day ago
I also doubt it is really that valuable on the OpenAI/Anthropic scale, at the same time if people will use it and it will work for them on the personal scale it is already a major win for me. New ideas and optimizations I could never have thought of might bring this up from a toy model to an actually useful model trained locally. Then people could add RL and RLHF and other cool things to it to make it even better.
Comment by bubblegumcrisis 1 day ago
Comment by bubblegumcrisis 1 day ago
Comment by volotat 23 hours ago
Comment by bubblegumcrisis 20 hours ago
The first starts with: Thanks for your response. The second starts with: Hey, do me a favor And the third starts with: I had no idea HN censors
If you can see all three - then you can also see the censorship by starting a private browser and looking at this story.
There will only be the second and the third.
I've tested in private sessions on firefox and safari and through a few different ip proxies.
I'm going to do some testing in other stories/comments to see whether they detect words, or general sentiment. And I'll get some statistics on if this is predictable censorship, a one time deal, or whether I just hit a race condition in their code somewhere.
Comment by nextaccountic 14 hours ago
Maybe hit up hn@ycombinator.com and ask to remove your shadowban of sorts. Idk how effective is that.
Anyway something more concerning is that you have at least two flagged comments (in the first page at least). Flagged comments means your comments violated HN rules, at least according to other HN users. It's hard to understand the HN social etiquette but in short, if you are being mean to other people, you might get flagged. So it's largely about the tone of your comments rather than their substance. You may want to read the guidelines (the section "in comments") here https://news.ycombinator.com/newsguidelines.html
Comment by bubblegumcrisis 11 hours ago
Comment by bubblegumcrisis 11 hours ago
So strange.
I had this weird epiphany this year - you know the Jan 6th - you know how the republicans refused to impeach Trump. I just couldn't wrap my head around the fact they would let someone potentially trying to kill them off the hook.
And then it dawned on me - they were in on it. If they were in on it, everything makes perfect sense. It's the simplest solution.
But it was such a leap for me.
All of the "obviously paid for bot comments" on this site. I always thought - they just slip through - but actually, the simplest solution is that this site is propagating a narrative.
So weird. And so gross. Where did the idealism of the tech industry go? "Don't be evil," they said.
Comment by bubblegumcrisis 1 day ago