我们知道线性回归一般都用来解决回归类问题,例如房价预测,气温预测等。实际上,加上 Softmax 这样的技术,我们还可以使用线性回归来解决多分类问题。Softmax 是对网络结构中输出层的改造,其示意图如下:
Softmax 技术细节
上图中,x1、x2 和 x3 是输入层,它们分别通过两个线性回归模型来产生 y1 和 y2:
接着,把 y1 和 y2 输入到 Softmax 模块中,输出 y1' 和 y2',经过 Softmax 处理后的结果,是一个离散概率分布,即 y1' + y2' = 1。于是,我们可以用 y1' 和 y2' 来表示不同分类的预测概率。