Java反向传播算法非常慢

时间:2016-02-02 18:41:16

标签: java algorithm backpropagation

我有一个大问题。我尝试创建一个神经网络,并希望用反向传播算法训练它。我在这里找到了这个教程http://mattmazur.com/2015/03/17/a-step-by-step-backpropagation-example/,并试图用Java重新创建它。当我使用他使用的训练数据时,我会得到与他相同的结果。 没有反向传播,我的TotalError几乎与他的相同。当我像他一样使用后向后传播10 000次时,我得到了几乎相同的错误。但是他使用了2个输入神经元,2个隐藏神经元和2个输出但是我想将这个神经网络用于OCR,所以我需要更多的神经元。但是,如果我使用例如49输入神经元,49隐藏神经元和2输出神经元,更改权重需要很长时间以获得小错误。 (我相信它需要永远.....)。我的学习率为0.5。在我的网络构造函数中,我生成神经元并给它们提供与教程中相同的训练数据,并用更多的神经元测试它,我给它们随机的权重,输入和目标。所以我不能将它用于许多神经元,它需要很长时间或者我的代码有问题吗?我应该提高学习率,偏见或开始体重吗? 希望你能帮助我。

package de.Marcel.NeuralNetwork;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Random;

public class Network {
    private ArrayList<Neuron> inputUnit, hiddenUnit, outputUnit;

    private double[] inHiWeigth, hiOutWeigth;
    private double hiddenBias, outputBias;

    private double learningRate;

    public Network(double learningRate) {
        this.inputUnit = new ArrayList<Neuron>();
        this.hiddenUnit = new ArrayList<Neuron>();
        this.outputUnit = new ArrayList<Neuron>();

        this.learningRate = learningRate;

        generateNeurons(2,2,2);

        calculateTotalNetInputForHiddenUnit();
        calculateTotalNetInputForOutputUnit();
    }

    public double calcuteLateTotalError () {
        double e = 0;
        for(Neuron n : outputUnit) {
            e += 0.5 * Math.pow(Math.max(n.getTarget(), n.getOutput()) - Math.min(n.getTarget(), n.getOutput()), 2.0);
        }

        return e;
    }

    private void generateNeurons(int input, int hidden, int output) {
        // generate inputNeurons
        for (int i = 0; i < input; i++) {
            Neuron neuron = new Neuron();

            // for testing give each neuron an input
            if(i == 0) {
                neuron.setInput(0.05d);
            } else if(i == 1) {
                neuron.setOutput(0.10d);
            }

            inputUnit.add(neuron);
        }

        // generate hiddenNeurons
        for (int i = 0; i < hidden; i++) {
            Neuron neuron = new Neuron();

            hiddenUnit.add(neuron);
        }

        // generate outputNeurons
        for (int i = 0; i < output; i++) {
            Neuron neuron = new Neuron();

            if(i == 0) {
                neuron.setTarget(0.01d);
            } else if(i == 1) {
                neuron.setTarget(0.99d);
            }

            outputUnit.add(neuron);
        }

        // generate Bias
        hiddenBias = 0.35;
        outputBias = 0.6;

        // generate connections
        double startWeigth = 0.15;
        // generate inHiWeigths
        inHiWeigth = new double[inputUnit.size() * hiddenUnit.size()];
        for (int i = 0; i < inputUnit.size() * hiddenUnit.size(); i += hiddenUnit.size()) {
            for (int x = 0; x < hiddenUnit.size(); x++) {
                int z = i + x;
                inHiWeigth[z] = round(startWeigth, 2, BigDecimal.ROUND_HALF_UP);

                startWeigth += 0.05;
            }
        }

        // generate hiOutWeigths
        hiOutWeigth = new double[hiddenUnit.size() * outputUnit.size()];
        startWeigth += 0.05;
        for (int i = 0; i < hiddenUnit.size() * outputUnit.size(); i += outputUnit.size()) {
            for (int x = 0; x < outputUnit.size(); x++) {
                int z = i + x;
                hiOutWeigth[z] = round(startWeigth, 2, BigDecimal.ROUND_HALF_UP);

                startWeigth += 0.05;
            }
        }
    }

    private double round(double unrounded, int precision, int roundingMode)
    {
        BigDecimal bd = new BigDecimal(unrounded);
        BigDecimal rounded = bd.setScale(precision, roundingMode);
        return rounded.doubleValue();
    }

    private void calculateTotalNetInputForHiddenUnit() {
        // calculate totalnetinput for each hidden neuron
        for (int s = 0; s < hiddenUnit.size(); s++) {
            double net = 0;
            int x = (inHiWeigth.length / inputUnit.size());

            // calculate toAdd
            for (int i = 0; i < x; i++) {
                int v = i + s * x;
                double weigth = inHiWeigth[v];
                double toAdd = weigth * inputUnit.get(i).getInput();
                net += toAdd;
            }

            // add bias
            net += hiddenBias * 1;
            net = net *-1;
            double output =  (1.0 / (1.0 + (double)Math.exp(net)));
            hiddenUnit.get(s).setOutput(output);
        }
    }

    private void calculateTotalNetInputForOutputUnit() {
        // calculate totalnetinput for each hidden neuron
        for (int s = 0; s < outputUnit.size(); s++) {
            double net = 0;
            int x = (hiOutWeigth.length / hiddenUnit.size());

            // calculate toAdd
            for (int i = 0; i < x; i++) {
                int v = i + s * x;
                double weigth = hiOutWeigth[v];
                double outputOfH = hiddenUnit.get(s).getOutput();
                double toAdd = weigth * outputOfH;
                net += toAdd;
            }

            // add bias
            net += outputBias * 1;
            net = net *-1;
            double output = (double) (1.0 / (1.0 + Math.exp(net)));
            outputUnit.get(s).setOutput(output);
        }
    }

    private void backPropagate() {
        // calculate ouputNeuron weigthChanges
        double[] oldWeigthsHiOut = hiOutWeigth;
        double[] newWeights = new double[hiOutWeigth.length];
        for (int i = 0; i < hiddenUnit.size(); i += 1) {
            double together = 0;
            double[] newOuts = new double[hiddenUnit.size()];
            for (int x = 0; x < outputUnit.size(); x++) {
                int z = x * hiddenUnit.size() + i;
                double weigth = oldWeigthsHiOut[z];
                double target = outputUnit.get(x).getTarget();
                double output = outputUnit.get(x).getOutput();

                double totalErrorChangeRespectOutput = -(target - output);
                double partialDerivativeLogisticFunction = output * (1 - output);
                double totalNetInputChangeWithRespect = hiddenUnit.get(x).getOutput();
                double puttedAllTogether = totalErrorChangeRespectOutput * partialDerivativeLogisticFunction
                        * totalNetInputChangeWithRespect;
                double weigthChange = weigth - learningRate * puttedAllTogether;

                // set new weigth
                newWeights[z] = weigthChange;
                together += (totalErrorChangeRespectOutput * partialDerivativeLogisticFunction * weigth);
                double out = hiddenUnit.get(x).getOutput();
                newOuts[x] = out * (1.0 - out);
            }
            for (int t = 0; t < newOuts.length; t++) {
                inHiWeigth[t + i] = (double) (inHiWeigth[t + i] - learningRate * (newOuts[t] * together * inputUnit.get(t).getInput()));
            }
            hiOutWeigth = newWeights;
        }
    }
}

我的神经元课程:

package de.Marcel.NeuralNetwork;

public class Neuron {
    private double input, output;
    private double target;

    public Neuron () {

    }

    public void setTarget(double target) {
        this.target = target;
    }

    public void setInput (double input) {
        this.input = input;
    }

    public void setOutput(double output) {
        this.output = output;
    }

    public double getInput() {
        return input;
    }

    public double getOutput() {
        return output;
    }

    public double getTarget() {
        return target;
    }
}

2 个答案:

答案 0 :(得分:2)

想一想:通过49-> 49-> 2神经元进行10,000次传播。在输入层和隐藏层之间,您有49 * 49个链接要传播,因此部分代码正在执行大约2400万次(10,000 * 49 * 49)。这需要时间。您可以尝试100次传播,看看需要多长时间,只是为了给您一个想法。

可以采取一些措施来提高性能,例如使用普通数组而不是使用ArrayList,但这对于Code Review网站来说是一个更好的主题。此外,不要指望这会带来巨大的改善。

答案 1 :(得分:1)

您的反向传播代码具有O(h * o + h ^ 2)* 10000的复杂度,其中h是隐藏神经元的数量,o是输出神经元的数量。这就是原因。

你有一个循环,可以为所有隐藏的神经元执行...

for (int i = 0; i < hiddenUnit.size(); i += 1) {

...包含另一个为所有输出神经元执行的循环...

for (int x = 0; x < outputUnit.size(); x++) {

...还有一个额外的内循环,它会为所有隐藏的神经元再次执行...

double[] newOuts = new double[hiddenUnit.size()];
for (int t = 0; t < newOuts.length; t++) {

...并执行所有万次。添加此O(i + h + o)[初始对象创建] + O(i * h + o * h)[初始权重] + O(h * i)[计算净输入] + O(h * o)[计算净产出]。

难怪它永远消失;您的代码充斥着嵌套循环。如果您希望它更快,请将这些因素考虑在内 - 例如,组合对象创建和初始化 - 或减少神经元的数量。但是,显着减少反向传播呼叫的数量是使这种运行更快的最好方法。