Adapting AlphaEvolve to Optimize Fully Homomorphic Encryption on TPUs
Dieses Papier stellt AlphaEvolve vor, ein von einem LLM gesteuertes evolutionäres Suchsystem, das die gemeinsame Optimierung von Fully Homomorphic Encryption-Kernen für Google TPUs automatisiert und im Vergleich zu den fortschrittlichsten menschlich entwickelten Implementierungen sowohl für TFHE- als auch für CKKS-Schemata signifikante Verbesserungen der Latenz erzielt.
Originalarbeit lizenziert unter CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Dies ist eine KI-generierte Erklärung des untenstehenden Papers. Sie wurde nicht von den Autoren verfasst oder gebilligt. Für technische Genauigkeit konsultieren Sie das Originalpaper. Vollständigen Haftungsausschluss lesen
Stellen Sie sich vor, Sie haben einen unglaublich leistungsstarken, hochgeschwindigkeitsfähigen Rennwagen (die TPU, ein spezieller Computerchip von Google). Sie besitzen auch ein sehr komplexes, geheimes Rezept für einen Kuchen, das gebacken werden muss, während die Zutaten noch in einem Safe verschlossen sind (dies ist Fully Homomorphic Encryption oder FHE, die es Computern ermöglicht, Mathematik auf geheimen Daten durchzuführen, ohne die Daten selbst je zu sehen).
Das Problem? Das Rezept ist so kompliziert und die Anweisungen so starr, dass der Rennwagen, wenn Sie versuchen, den Kuchen damit zu backen, 90 % seiner Zeit nur damit verbringt, darauf zu warten, dass die Zutaten umgelagert werden. Es ist, als würde man versuchen, einen Formel-1-Wagen durch eine Nachbarschaft zu fahren, in der an jedem Block ein Stoppschild steht. Der Wagen ist schnell, aber der Verkehr (die Softwareanweisungen) ist langsam.
Normalerweise versuchen menschliche Ingenieure, dies zu beheben, indem sie die Anweisungen manuell umschreiben, verschiedene Wege ausprobieren, die Zutaten zu bewegen, und prüfen, ob der Wagen schneller wird. Es ist ein langsamer Prozess des „Versuchens und Prüfens", und oft verpassen sie den besten Weg zu fahren, weil die Anweisungen hinter Schichten verwirrender Software verborgen sind.
Hier kommt AlphaEvolve ins Spiel: Der KI-Rennwagen-Tuner
Dieser Artikel beschreibt ein neues Werkzeug namens AlphaEvolve. Stellen Sie es sich als einen KI-Rennwagen-Tuner vor, der nicht nur den Motor justiert, sondern das gesamte Fahrhandbuch neu schreibt, um es perfekt auf die spezifische Strecke abzustimmen.
So funktioniert es, unter Verwendung einfacher Analogien:
1. Die „evolutionäre" Garage
Anstatt dass ein einzelner menschlicher Ingenieur rät, agiert AlphaEvolve wie ein Schwarm aus Tausenden winziger, unermüdlicher Mechaniker.
- Der Prozess: Es beginnt mit einem guten Rezept (dem ursprünglichen Code). Dann erstellt es Hunderte leicht unterschiedlicher Versionen dieses Rezepts.
- Die Probefahrt: Es führt jede Version am tatsächlichen Rennwagen (der TPU-Hardware) aus.
- Die Bewertung: Es misst genau, wie lange das Backen des Kuchens dauert.
- Die Auswahl: Wenn eine Version langsamer ist oder den Kuchen ruiniert (einen Mathetest nicht besteht), wird sie in den Müll geworfen. Wenn sie schneller ist, wird sie behalten.
- Die Evolution: Die KI nimmt die „gewinnenden" Versionen, mischt sie zusammen und erstellt neue, noch bessere Versionen. Sie wiederholt diesen Zyklus immer wieder, wie bei der natürlichen Selektion, bis sie den absolut schnellsten Weg findet, den Kuchen zu backen.
2. Der Sicherheitswächter (Kritisch für Geheimnisse)
Da dies geheime Daten betrifft, kann die KI nicht einfach „halluzinieren" oder neue Regeln erfinden, die versehentlich das Geheimnis preisgeben könnten.
- Der Türsteher: Bevor ein neues Rezept auf dem echten Rennwagen ausgeführt werden darf, prüft es ein strenger „Türsteher". Er fragt: „Erzeugt dieses Rezept immer noch exakt denselben Kuchen wie das Original? Ist das Geheimnis noch sicher?"
- Wenn die KI versucht zu betrügen oder die Mathematik so zu verändern, dass die Sicherheit gebrochen wird, lehnt der Türsteher dies sofort ab. Dies stellt sicher, dass die KI nur Geschwindigkeit findet, keine Sicherheitslücken.
3. Was hat die KI entdeckt?
Nachdem dieser Prozess nur 24 Stunden lang durchgeführt wurde, fand die KI Lösungen, die menschliche Experten übersehen hatten. Hier sind die „Aha!"-Momente, die sie hatte:
- Der „Batching"-Trick (Loop Unrolling): Stellen Sie sich einen Arbeiter vor, der eine schwere Kiste tragen, zum Ofen laufen, sie hineinstellen, zurücklaufen, eine weitere Kiste holen und dies wiederholen muss. Die KI erkannte: „Warum hin und her laufen? Lassen Sie uns 8 Kisten auf einmal greifen und alle in einer einzigen Fahrt tragen." Durch die Umorganisation des Datengriffs sparte es enorme Mengen an Zeit.
- Der „Keine-Änderung"-Trick (Entfernen von Typumwandlungen): Manchmal sagte das ursprüngliche Rezept: „Nehmen Sie diese Zutat, ändern Sie ihr Etikett von 'Gramm' in 'Unzen', und ändern Sie es vor dem Kochen wieder zurück in 'Gramm'." Die KI erkannte, dass dies Zeitverschwendung war. Die Zutat war bereits die richtige Größe; die Etikettenänderung war unnötig. Durch das Löschen dieses sinnlosen Schritts wurde der Prozess viel schneller.
- Der „Perfekte Sitz"-Trick (Tiling): Der Rennwagen hat Regale in bestimmten Größen (Speicherregister). Das ursprüngliche Rezept versuchte, ein riesiges Brot auf ein kleines Regal zu legen, was ein Chaos verursachte. Die KI fand heraus, wie man das Brot in perfekte, kleinere Stücke schneidet, die genau in die Regale passen, wodurch der Wagen seine volle Kraft nutzen kann, ohne Platz zu verschwenden.
Die Ergebnisse
Indem man der KI das automatische „Versuchen und Prüfen" überließ:
- TFHE (Der „Geheim-Mathematik"-Kuchen): Die Zeit zum Backen des Kuchens sank um den Faktor 2,5.
- CKKS (Der „Datenanalyse"-Kuchen): Die Zeit zum Backen sank um den Faktor 1,3.
Das große Ganze
Der Artikel kommt zu dem Schluss, dass Menschen zwar hervorragend darin sind, die ursprünglichen Geheimrezepte zu entwerfen, aber oft zu langsam darin sind, den effizientesten Weg zu finden, den Rennwagen zu fahren. Indem wir eine KI einsetzen, die Tausende von Variationen an einem Tag testen kann, während sie streng überprüft, dass die Geheimnisse sicher bleiben, können wir die volle Geschwindigkeit moderner Computerchips für Privatsphäre-Technologien freisetzen.
Kurz gesagt: Die KI hat keine neue Art erfunden, den Kuchen zu backen; sie hat nur einen viel schnelleren Weg gefunden, die Zutaten zum Ofen zu tragen.
Ertrinken Sie in Arbeiten in Ihrem Fachgebiet?
Erhalten Sie tägliche Digests der neuesten Arbeiten passend zu Ihren Forschungsbegriffen — mit technischen Zusammenfassungen, in Ihrer Sprache.