多个 softmax 分类(Keras)

Tac*_*tis 5 python neural-network conv-neural-network keras

我正在尝试构建一个输出 2 个标签的 CNN,其中每个标签有 12 种可能性;输入是一个图像。

换句话说,我想要的输出是 A(12 个选项中的一个)和 B(12 个不同选项中的一个)。

单独使用每个标签,首选当然是 softmax,但我找不到是否可以做这样的“多个 softmax”标签。

显然,我可以使用 sigmoid 输出输出 24 个神经元,然后手动归一化,但是网络不会本质上耦合每组中的 12 个选项。

我正在编写我想要的伪伪代码,如果它存在的话:

model = Sequential()
model.add(Convolution2D(64, kernel_size=3,activation='relu', input_shape=image_shape))
model.add(Flatten())
model.add(Dense(256,activation='relu'))
# pseudo-pseudo-code
model.add(Dense((12,12),activation=('softmax','softmax'))) # <- here is where I would have liked 2 softmax outputs with 12 neurons in each
Run Code Online (Sandbox Code Playgroud)

欢迎任何解决方案(我正在使用 Keras,但会很高兴使用任何其他包的解决方案,甚至自己编写代码)。

编辑:我还可以生成代表所有组合的 144 个输出,但我不确定这是否是一个好的解决方案 - 对此的任何评论也非常受欢迎

小智 5

我会使用带有 2 个输出的功能 API。https://keras.io/getting-started/functional-api-guide/#multi-input-and-multi-output-models 中的指南

一些关于:

input = Input(input_shape=image_shape)
cnv = Convolution2d(...)(input)
hidden = Dense(256, ...)(Flatten()(cnv))
output1 = Dense(12, activation='softmax')(hidden)
output2 = Dense(12, activation='softmax')(hidden)
model = keras.models.Model(inputs=input, outputs=[output1, output2])
model.compile(optimizer='sgd', loss=['binary_crossentropy', 'binary_crossentropy'],
              loss_weights=[1., 1.])
Run Code Online (Sandbox Code Playgroud)

  • 以防万一有人来读这个,`loss` 应该是 `categorical_crossentropy` 而不是二进制 (4认同)