アーカイブ / Computer Vision
画像特徴を分けて処理するPartialNet:畳み込みと注意機構を並列に使う
PartialNet
PartialNetの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 3 : The overall architecture of our PartialNet, consisting of four hierarchical stages, each incorporating a series of PartialNet blocks followed by an embedding or merging layer. The last three layers are dedicated to feature classification. Where ⊙ \odot and ⊗ \otimes denote element-wise multiplication and matrix multiplication respectively.
概要
- PATConvは特徴チャネルを分け、畳み込みと視覚的注意を並列に適用して結合する。
- DPConvは計算量への制約を設けながら、層ごとの分割比率を学習する。
- 著者らのImageNet-1K評価では、PartialNet-T2のTop-1正解率は80.2%、FasterNet-T2は78.9%だった。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、PartialNetの仕組み
- 01入力画像
- 02特徴チャネルを分割
- 03畳み込みと注意
- 04特徴を結合
- 05分類結果
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
PartialNetは、画像から取り出した数値表現である特徴を分け、異なる処理に振り向ける画像認識モデルだ。チャネルは特徴の種類に対応するまとまりを指す。
画像認識では、正解率に加えて計算量や判定速度も実用上の課題になる。著者らは特徴チャネル間の重複に着目し、全チャネルへ同じ処理を施す代わりに、部分ごとに役割を割り当てる方法を検討した。
手法
画像から得た特徴マップをチャネル単位で分け、一方に畳み込み、他方に視覚的注意を並列に適用して結合する。畳み込みは近くの画像情報を組み合わせる処理、注意機構は特徴の重みを調整する処理だ。PATConvではチャネル、画像内の位置、離れた位置同士に着目する3種類の注意を使う。
たとえば製品画像を分類する場合、入力画像から作った特徴を分けて処理し、結合した特徴から種類を判定する流れになる。DPConvは学習中に層ごとの分割比率を調整し、計算量にも制約を設ける。PartialNetは4段の構成を取り、離れた位置の関係を扱う自己注意を最終段で使う。
新規性
著者らが提案するPATConvは、特徴チャネルを分け、一方に畳み込み、他方に視覚的注意を並列に適用して結合する。注意機構を全チャネルへ適用する方法とは、処理する範囲が異なる。
さらにDPConvで層ごとの分割比率を学習する。チャネル注意、空間注意、自己注意を組み込んだ構成を示し、自己注意は最終段に配置している。
従来手法との違い
FasterNetの部分畳み込みは、一部のチャネルを処理し、残りをそのまま通す。PartialNetのPATConvは分けたチャネルに畳み込みと注意機構をそれぞれ適用する。FasterNetが標準設定で分割比率を4分の1に固定するのに対し、DPConvは層ごとに比率を学習する。
著者らのImageNet-1Kの表では、T2同士ならPartialNetの正解率と処理枚数が高い。一方、T0とT1のV100での処理枚数はFasterNetを下回る。速度差はモデルの規模と測定環境を指定して読む必要がある。
実験結果
著者らはImageNet-1Kの約130万枚で300エポック学習し、5万枚で分類を検証した。PartialNet-T2とFasterNet-T2のTop-1正解率は80.2%対78.9%。バッチサイズ256での処理枚数は、MI250で毎秒4761対4189枚、V100で3074対2455枚だった。
COCO 2017では、事前学習モデルをMask R-CNNの画像特徴抽出部分に使い、11万8000枚で12エポック学習、5000枚で検証した。著者らの表でPartialNet-SとFasterNet-SのAPは、物体検出で42.7対39.9、個体領域分割で39.3対36.9。APは精度と検出漏れの関係をまとめる指標だ。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案:工場で撮影した製品画像から、外観上の不良候補を分類する工程に試す。入力は既設カメラの画像、出力は良品または不良種別の候補とし、担当者が確認する画面へ送る。撮影画像に加え、良品と不良種別を示す学習用ラベルが必要だ。
既設カメラ以外のセンサーを前提とせず、撮影条件に合わせた画像の形式調整と分類器の追加学習を行う。論文の評価は一般画像の分類などであり、この工程での精度や速度は別途確かめる必要がある。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案:対象工程の画像を既設カメラで集め、良品と不良種別のラベルを付ける。撮影位置や照明条件を記録し、学習用と評価用を分ける。PartialNetと現在の分類器を、同じ画像とラベルを使って対象工程向けに学習する。
画像形式を各分類器の入力に合わせたうえで、評価用画像の種別ごとの見逃し数と誤判定数を比べる。実際の処理環境で1枚当たりの判定時間と一定時間の処理枚数も測り、論文の一般画像での結果が対象工程に当てはまるか判断する。
限界と課題
著者らは、畳み込み側の分割比率が大きすぎると注意機構が働きにくく、小さすぎると局所的な特徴を捉えにくいと説明する。自己注意の計算量は特徴マップの大きさに対して二乗で増えるため、最終段に限っている。DPConvの説明には、各層のチャネル数を2の累乗にする条件もある。
編集上の検証事項は、対象画像での誤分類、実行環境での判定時間と処理枚数、追加学習後の分割比率だ。確認できた本文の範囲では、工場の外観検査に適用した結果は示されていない。
出典
元のタイトル・要旨を確認する(英語)
Partial Channel Network: Compute Fewer, Perform Better
Designing a module or mechanism that enables a network to maintain low parameters and FLOPs without sacrificing accuracy and throughput remains a challenge. To address this challenge and exploit the redundancy within feature map channels, we propose a new solution: partial channel mechanism (PCM). Specifically, through the split operation, the feature map channels are divided into different parts, with each part corresponding to different operations, such as convolution, attention, pooling, and identity mapping. Based on this assumption, we introduce a novel partial attention convolution (PATConv) that can efficiently combine convolution with visual attention. Our exploration indicates that the PATConv can completely replace both the regular convolution and the regular visual attention while reducing model parameters and FLOPs. Moreover, PATConv can derive three new types of blocks: Partial Channel-Attention block (PAT_ch), Partial Spatial-Attention block (PAT_sp), and Partial Self-Attention block (PAT_sf). In addition, we propose a novel dynamic partial convolution (DPConv) that can adaptively learn the proportion of split channels in different layers to achieve better trade-offs. Building on PATConv and DPConv, we propose a new hybrid network family, named PartialNet, which achieves superior top-1 accuracy and inference speed compared to some SOTA models on ImageNet-1K classification and excels in both detection and segmentation on the COCO dataset. Our code is available at https://github.com/haiduo/PartialNet.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01




