アーカイブ / Generative AI
分類モデルKANの層と接続を、遺伝的アルゴリズムで自動探索する
GA-KAN
GA-KANの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Fig. 1: The overall framework of the proposed method, where t t denotes the generation, P t P_{t} represents the parent population, and Q t Q_{t} represents the offspring population. The next parent population, P t + 1 P_{t+1} , is selected based on a combination of P t P_{t} and Q t Q_{t} . The dashed box indicates an element, while the solid box highlights a key step in GA.
概要
- KANの層数、接続、格子値をビット列にまとめ、遺伝的アルゴリズムで構造を探索する。
- 著者らは五つの分類データセットで従来手法より良い性能と、標準KANより少ないパラメータ数を報告した。
- 実用時は分類精度とともに、探索時間と得られた式の安定性を測る必要がある。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、GA-KANの仕組み
- 01分類用の学習データ
- 02KAN構造を符号化
- 03交叉・突然変異で探索
- 04選ばれたKAN
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
GA-KANは、分類に使うKANの構造を自動で探す。KANはニューロン間の接続に学習可能な曲線を置くネットワークで、入力と出力の関係を数式として表す余地がある。一方、層の数や接続、曲線の設定を人が調整する作業は重い。
分類の精度に加え、使うパラメータ数と説明のしやすさを検討したい場面では、構造を決める過程も重要になる。本研究は、接続を間引いたKANまで探索対象に含め、調整作業を減らそうとする。
手法
最初に、入力と出力の項目数、最大層数、各層の最大ニューロン数を決める。層間の接続を0か1、曲線の格子値と層数もビット列に記録する。このビット列を「染色体」と呼び、ゼロマスクと縮退処理で不要な層を外して、実際のKANへ戻す。
候補ごとに学習データで一定回数学習し、検証データでの損失の最小値を評価に使う。遺伝的アルゴリズムとは、良い候補を選び、染色体の一部を交換する交叉や、値を反転する突然変異を繰り返す探索法である。例えば測定値から種類を分類する課題なら、複数の接続パターンを試し、選ばれた構造と予測結果を出す。
新規性
GA-KANは、KANの層数、ニューロン間の接続、曲線を表す格子値を一つの染色体に符号化し、進化的な探索で同時に選ぶ。著者らは、進化計算によるKAN構造の自動最適化を初めて試みたと主張する。
疎な接続も探索対象にするため、全てのニューロンを結ぶ構造に固定されない。固定長の染色体から異なる深さのKANを取り出すために、層全体を無効化するゼロマスクと縮退処理を組み合わせる。
従来手法との違い
元のKANでは、課題に合わせた構造設計や、不要な接続を削る際の設定に人の調整が要る。GA-KANは接続の有無と深さを探索し、検証データでの損失を使って候補を選ぶ。比較対象には標準KANと従来の分類手法が含まれる。
著者らは、五つの分類データセットすべてで標準KANよりパラメータ数を減らし、従来手法より良い成績を得たと述べる。ただし、手動調整の削減と探索に要する計算量は別の論点である。
実験結果
著者らは二つの玩具データセットと五つの分類データセットで検証した。玩具データでは元のKANに必要だった手動調整なしで最適な結果を得たと報告し、五つの分類データセットでは従来手法を上回り、標準KANよりパラメータ数を減らしたと述べる。
WineとIrisについては、解釈に使える記号式を示した。確認できた本文の範囲では、各データセットの精度表や削減率まで確認できない。候補KANの評価手順には、20回の学習ステップで得る最小の検証損失を使うことが記されている。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案:製造工程で得た数値の検査記録から、製品の判定区分を予測する小規模な分類器を試す。入力は項目ごとの測定値と既存の判定ラベル、出力は判定区分、採用された入力項目、候補となる式である。
担当者は式と重要項目を検査手順に照らして確認する。式が出たことだけで原因が特定できたとは扱わず、予測が外れた記録と、測定条件が変わった記録を別に点検する。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案:数値項目を持つ小さな分類データを学習用、検証用、最終評価用に分け、GA-KANと標準KANを同じ入力で比べる。探索範囲と試行回数を記録し、最終評価用データは構造選択に使わない。
正解率だけでなく、クラス別の見逃し、パラメータ数、探索を含む所要時間を測る。得られた式については担当者が入力項目との対応を読み、異なるデータ分割でも同じ項目が残るか確認する。
限界と課題
著者らが問題として挙げるのは、従来KANの手動調整と、広いパラメータ空間である。GA-KANでも探索範囲の最大深さ、各層の最大ニューロン数、集団サイズなどは事前に指定するため、設計作業が完全に消えるわけではない。
編集上は、探索全体の所要時間と計算量、乱数を変えた際の安定性を確かめたい。抽出できる式が人に理解しやすいか、未知のデータでも成績が続くかも、用途ごとに評価する必要がある。
出典
元のタイトル・要旨を確認する(英語)
A Genetic Algorithm-Based Approach for Automated Optimization of Kolmogorov-Arnold Networks in Classification Tasks
To address the issue of interpretability in multilayer perceptrons (MLPs), Kolmogorov-Arnold Networks (KANs) are introduced in 2024. However, optimizing KAN structures is labor-intensive, typically requiring manual intervention and parameter tuning. This paper proposes GA-KAN, a genetic algorithm-based approach that automates the optimization of KANs, requiring no human intervention in the design process. To the best of our knowledge, this is the first time that evolutionary computation is explored to optimize KANs automatically. Furthermore, inspired by the use of sparse connectivity in MLPs in effectively reducing the number of parameters, GA-KAN further explores sparse connectivity to tackle the challenge of extensive parameter spaces in KANs. GA-KAN is validated on two toy datasets, achieving optimal results without the manual tuning required by the original KAN. Additionally, GA-KAN demonstrates superior performance across five classification datasets, outperforming traditional methods on all datasets and providing interpretable symbolic formulae for the Wine and Iris datasets, thereby enhancing model transparency. Furthermore, GA-KAN significantly reduces the number of parameters over the standard KAN across all the five datasets. The core contributions of GA-KAN include automated optimization, a new encoding strategy, and a new decoding process, which together improve the accuracy and interpretability, and reduce the number of parameters.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-30




