Global Average Pooling(GAP)の実現方法について

unread,

Jan 10, 2018, 9:56:37 PM1/10/18

to Chainer Japanese User Group

Global Average Pooling層 ( https://arxiv.org/pdf/1312.4400.pdf の3.2 )はChainerのFunctionsとしては実装されていないようです。

chainer.functions.average_pooling_2d のksizeを前の層の各チャネルのH x Wに合わせて指定すれば代用ができそうですが、

もし他に良い方法があれば共有ください。

unread,

Jan 11, 2018, 7:53:28 AM1/11/18

to Chainer Japanese User Group

meanあるいはaverageはどうですか。どちらもaxis=1, keepdims=Trueを指定することで想定しているGlobal Average Poolingと同じ演算ができるかと思います。
https://docs.chainer.org/en/stable/reference/generated/chainer.functions.average.html#chainer.functions.mean

2018年1月11日木曜日 11時56分37秒 UTC+9 Keita Shimizu:

unread,

Jan 16, 2018, 4:32:29 AM1/16/18

to Chainer Japanese User Group

ありがとうございます！

2018年1月11日木曜日 21時53分28秒 UTC+9 日暮大輝:

unread,

Jan 16, 2018, 8:38:44 AM1/16/18

to Chainer Japanese User Group

先の回答、axisの引数が間違っていました。axis=(2, 3)ですね。失礼しました。