Smol2Operator: Post-Training GUI Agents for Computer Use

Smol2Operator: Post-Training GUI Agents for Computer Use

なぜGUI Agentの訓練は難しいのか?Smol2Operatorが解決した「データの統一」問題 🧩

GUI操作AIの研究で大きな壁になっているのが、訓練データのフォーマットがバラバラという問題です。Smol2Operatorの論文で特に注目すべきは、この問題に正面から取り組んだ「Action Space Converter(アクションスペース変換器)」の開発です。

ここでいう「Action Space(アクションスペース)」とは、AIが実行できる操作の種類とその表現方法のことです。人間でいえば「語彙と文法」のようなもの。

現状のGUI操作データセットには、こんな課題があります:

→ データセットAでは座標が画像の左上を(0,0)とする絶対座標で表現
→ データセットBでは画面全体を0〜1に正規化した相対座標で表現
→ データセットCではUI要素のIDを使った抽象的な操作で表現
→ クリック、ドラッグ、テキスト入力などの基本操作さえ、表記方法がデータセットごとにバラバラ

これは例えるなら、同じ「家に帰る」という行動を、ある教科書では「go_home()」、別の教科書では「navigate(destination: “home”)」、また別では「move{target: house, action: enter}」のように書いているようなものです。人間なら意味が分かりますが、AIにとっては全く別の命令に見えてしまいます。

Smol2Operatorの研究チームは、この問題を解決するために統一アクションスペースを設計し、既存の主要なGUI操作データセットを全てこの形式に変換するツールを開発しました。これによって:

・複数のデータセットを組み合わせて訓練できるようになった(データ量が増えてモデルが賢くなる)
・異なる研究プロジェクトで訓練されたモデル同士が比較可能になった
・新しいデータセットを追加する際も、統一フォーマットに変換すれば簡単に利用できる

技術的には、Phase 1で「認識」(画面を見て理解する能力)を訓練し、Phase 2で「認知」(見た情報をもとに適切な操作を選ぶ能力)を訓練する2段階アプローチを採用しています。いきなり複雑な操作を教えるのではなく、まず「見る目」を養ってから「考える力」を鍛えるという、人間の学習プロセスに近い方法です。

画像解像度と座標システムの最適化実験も興味深いです。画質を上げれば細かいUI要素も認識しやすくなりますが、その分計算コストが増える。座標の表現方法も、絶対座標と相対座標それぞれに長所短所がある。こうしたトレードオフを丁寧に検証している点が、実用的なモデル開発に繋がっています。

この「データ統一」のアプローチは、GUI Agent研究だけでなく、他のAI分野でも参考になりそうです。結局のところ、AIの性能は訓練データの質と量に大きく依存するので、データを効率的に活用できる基盤を整えることが、研究の加速に不可欠なんですね。

オープンソースで全て公開されている点も素晴らしい。他の研究者がこの変換ツールを使って自分のデータセットを追加したり、改良したりできるので、コミュニティ全体で技術が進化していくはずです。

#機械学習 #データサイエンス #研究開発

https://huggingface.co/blog/smol2operator