CRISP: Pre-LLM Yet Text-Driven Visual Token Pruning for Efficient LVLM Inference
CRISP ist ein Pre-LLM, textgesteuertes Framework zur visuellen Token-Pruning, das eine zweistufige Pipeline nutzt, um instruktionsrelevante Token zu identifizieren und die semantische Diversität zu erhöhen, wobei es über 99,5 % der Genauigkeit beibehält und gleichzeitig die Inferenzlatenz für Large Vision-Language-Models um mehr als das Zweifache reduziert.
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 versuchen, ein Rätsel zu lösen, aber anstatt nur einiger weniger Hinweise erhalten Sie eine Bibliothek mit Millionen von Büchern, die alle gleichzeitig aufgeschlagen sind. Das ist im Wesentlichen die Arbeitsweise moderner „Large Vision-Language Models“ (LVLMs). Dies sind superintelligente Computerprogramme, die ein Bild betrachten und Fragen dazu beantworten können, wie ein digitaler Detektiv. Um dies zu tun, zerlegen sie das Bild in Tausende von winzigen digitalen Teilen, die „Tokens“ genannt werden. Betrachten Sie diese Tokens als einzelne Puzzleteile. Das Problem ist, dass der Computer versucht, jedes einzelne Stück des Puzzles zu lesen, selbst die, die nur blauen Himmel oder einen leeren Boden darstellen, noch bevor er überhaupt Ihre Frage betrachtet. Dies macht den Prozess unglaublich langsam und hungrig nach Rechenleistung – so, als würde man versuchen, eine ganze Enzyklopädie zu lesen, nur um herauszufinden, welche Farbe der Kragen des Katers hat.
Lange Zeit versuchten Wissenschaftler, dies zu beschleunigen, indem sie entweder Puzzleteile wahllos wegwarfen (was manchmal den Kragen des Katers löschte) oder indem sie warteten, bis der Computer bereits angefangen hatte, das ganze Buch zu lesen, bevor er entschied, was er überspringen sollte (was zu spät war, um Zeit zu sparen). Die große Frage war: Könnte man dem Computer beibringen, zuerst die Frage zu lesen und dann nur die spezifischen Puzzleteile auszuwählen, die benötigt werden, um sie zu beantworten, ohne dabei das gesamte System zu verlangsamen?
Hier kommt eine neue Methode namens CRISP ins Spiel. Betrachten Sie CRISP als einen super effizienten Bibliothekar, der vor dem eigentlichen Lesen des Detektivs arbeitet. Anstatt die ganze Bibliothek auf den Tisch zu werfen, hört CRISP zuerst auf Ihre Frage. Wenn Sie fragen: „Welche Farbe hat die Handtasche?“, scannt CRISP das Bild sofort und schnappt sich nur die Puzzleteile, die die Handtasche und die unmittelbare Umgebung zeigen. Aber es bleibt nicht dabei; es schnappt sich auch ein paar zusätzliche Teile, um sicherzustellen, dass der Hintergrundkontext (wie die Straße oder die Person, die sie hält) nicht verloren geht.
Die Forscher hinter CRISP fanden heraus, dass dieser „Erst fragen, dann schauen“-Ansatz viel klüger ist als die alten „Erst schauen, dann fragen“-Methoden. In ihren Tests zeigten sie, dass CRISP bis zu 88,9 % der visuellen Puzzleteile wegwerfen konnte (wodurch das Bild von Hunderten von Tokens auf nur noch 64 oder 128 reduziert wurde), während die Genauigkeit des Computers fast exakt dieselbe blieb, als hätte er das gesamte Bild betrachtet. Tatsächlich war es in einigen Tests so gut darin, die richtigen Hinweise zu finden, dass es sogar „Halluzinationen“ reduzierte – jene Momente, in denen eine KI Dinge erfindet, die gar nicht da sind.
Das Paper argumentiert explizit gegen zwei heute gängige Wege, dies zu tun. Ein Weg besteht darin, einfach die „wichtigsten“ aussehenden Teile eines Bildes auszuwählen, unabhängig von der Frage (text-agnostisch), was die Autoren als vergleichbar mit dem Starren auf ein Gemälde beschreiben, bevor man weiß, wonach man eigentlich sucht. Der andere Weg besteht darin, zu warten, bis der Computer tief in seinem Denkprozess steckt, um dann mit dem Löschen von Teilen zu beginnen, was die Autoren als Energieverschwendung bezeichnen, da der Computer die harte Arbeit des Lesens bereits erledigt hat. CRISP lehnt beides ab und beweist, dass man gleichzeitig schnell und klug sein kann, indem man das Pruning (das Beschneiden) vor dem schweren Denken durchführt.
Die Ergebnisse sind messbar und sehr spezifisch. Bei einem populären Modell namens LLaVA-1.5 behielt CRISP 99,5 % der ursprünglichen Leistung bei, während es nur 128 Tokens anstelle der üblichen 576 verwendete. Bei einem neueren Modell, LLaVA-NeXT, behielt es 96,9 % der Leistung mit nur 320 Tokens bei. Vielleicht am beeindruckendsten ist, dass diese Beschleunigung nicht nur die Genauigkeit bewahrte, sondern auch die Zeit bis zur Antwort um mehr als die Hälfte verkürzte. Die Autoren schlagen vor, dass diese Methode eine praktische, einsatzbereite Lösung ist, um diese intelligenten bildlesenden Computer viel schneller und kostengünstiger zu machen, insbesondere für Geräte, die nicht über massive Mengen an Energie verfügen.
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.