CNNで出力がほとんど同じになってた話

CNNで出力が偏っていた問題を少し解決したお話

CNNで5クラス分類をしてて、ずっと出力がおかしかった。

どんなふうにおかしかったのかというと、出力がとても偏っていた。

仮に、100個のデータがあり、均等に各クラス20個ずつだったとして

学習を進めるうちにCNNの出力は

クラス:出力数

class0: 1

class1: 90

class2: 0

class3: 0

class4: 9

みたいな形で、どんどん偏っていた。いろいろ調べてもわからなかったが、stack overflow でちょいちょい

”もしかしたら層数とかの中間特徴量が多いことは無いですか?”

という書き込みをみたので、それを試してみることに。

入力値は3ch信号の500×128 のFFTスペクトログラムデータ

元々はこれを

畳み込み1層目:128

畳み込み2層目:64

畳み込み3層目:32

畳み込み4層目:16

としていたが、変更後は

畳み込み1層目:10

畳み込み2層目:8

畳み込み3層目:5

と極端に減らして、かつ層数も1層減らした。

結果、うまく動くようになった。

結論

研究データの取り直しがほぼ確定\(^o^)/

tensorflow機械学習深層学習
スポンサーリンク
Toufuをフォローする

コメント

タイトルとURLをコピーしました