CNNで出力が偏っていた問題を少し解決したお話
CNNで5クラス分類をしてて、ずっと出力がおかしかった。
どんなふうにおかしかったのかというと、出力がとても偏っていた。
仮に、100個のデータがあり、均等に各クラス20個ずつだったとして
学習を進めるうちにCNNの出力は
クラス：出力数
class0: 1
class1: 90
class2: 0
class3: 0
class4: 9
みたいな形で、どんどん偏っていた。いろいろ調べてもわからなかったが、stack overflow でちょいちょい
”もしかしたら層数とかの中間特徴量が多いことは無いですか？”
という書き込みをみたので、それを試してみることに。
入力値は3ch信号の500×１２８ のFFTスペクトログラムデータ
元々はこれを
畳み込み１層目:128
畳み込み2層目:64
畳み込み3層目:32
畳み込み4層目:16
としていたが、変更後は
畳み込み１層目:10
畳み込み2層目:8
畳み込み3層目:5
と極端に減らして、かつ層数も１層減らした。
結果、うまく動くようになった。
結論
研究データの取り直しがほぼ確定＼(^o^)／
コメント