1

I have a dataset size of ~500000 with input dimension 46. I am trying to use Pybrain to train the network but the training is extremely slow for the whole dataset. Using batches of 50000 data points, each batch takes more than 2 hours for training.

What are caveats of optimizing the network design so that the training is faster?

sonaam1234
  • 59
  • 1
  • 5

3 Answers3

2

Here are some of the things that influence your training speed:

  • Number of weights in your network
  • Speed of your CPU
  • Package you are using (mostly engine it is working on, in PyLearn this is Theano)
  • If all your data fits in memory or you are reading from disk in between batches

With regards to network design the only thing you can really do is make the network more shallow to reduce the number of weights. To reduce the number of epochs there might be other options like adding residual connections but that will not decrease the training time of 1 epoch.

Without more information it is unclear where the bottleneck is, but 20 hours for one epoch seems a bit high. The easiest and biggest improvement you will be able to get is to use a good GPU, which should be possible using pylearn since it is built on top of Theano.

Jan van der Vegt
  • 9,328
  • 34
  • 52
  • Hi, Reading batched from the file one batch at a time rather than processing chunks from all data read in memory drastically reduced the time. Thanks! – sonaam1234 Aug 30 '16 at 07:34
1

Use some data set analysis (like Principal Component Analysis) trying to figure out the underlying structure of your data set. You could probably reduce the dimension without losing too much information.

ujhuyz0110
  • 111
  • 2
0

2 hours for a batch is too long even with a CPU. How many layers you have? How many neutrons you have? Have you tried a single layer? Try to reduce the complexity of your model.

You might want to train a softmax regression, just to get an lower bound for your computation power. If you find your machine struggle even with softmax regression, you'll need to upgrade your hardware.

ABCD
  • 3,510
  • 2
  • 18
  • 30
  • I was using a backprop trainer on a network with 5 layers (46, 30, 15, 5, 1) and I was using trainer.trainUntilConvergence(maxEpochs = 1000 ). I have changed the network to 3 layers (46, 20, 1) and maxEpochs to 100. – sonaam1234 Aug 30 '16 at 06:19
  • 5 layers is very computational intensive! Try 3 or even 1 to get a feeling. – ABCD Aug 30 '16 at 06:26