Slashdot Mirror


Google Open-Sources GPipe, a Library For Training Large Deep Neural Networks (venturebeat.com)

An anonymous reader quotes a report from VentureBeat: Google's AI research division today open-sourced GPipe, a library for "efficiently" training deep neural networks (layered functions modeled after neurons) under Lingvo, a TensorFlow framework for sequence modeling. It's applicable to any network consisting of multiple sequential layers, Google AI software engineer Yanping Huang said in a blog post, and allows researchers to "easily" scale performance. As Huang and colleagues explain in an accompanying paper ("GPipe: Efficient Training of Giant Neural Networks using Pipeline Parallelism"), GPipe implements two nifty AI training techniques. One is synchronous stochastic gradient descent, an optimization algorithm used to update a given AI model's parameters, and the other is pipeline parallelism, a task execution system in which one step's output is streamed as input to the next step.

Most of GPipe's performance gains come from better memory allocation for AI models. On second-generation Google Cloud tensor processing units (TPUs), each of which contains eight processor cores and 64 GB memory (8 GB per core), GPipe reduced intermediate memory usage from 6.26 GB to 3.46GB, enabling 318 million parameters on a single accelerator core. Without GPipe, Huang says, a single core can only train up to 82 million model parameters. That's not GPipe's only advantage. It partitions models across different accelerators and automatically splits miniature batches (i.e., "mini-batches") of training examples into smaller "micro-batches," and it pipelines execution across the micro-batches. This enables cores to operate in parallel, and furthermore accumulate gradients across the micro-batches, thereby preventing the partitions from affecting model quality.

22 comments

  1. The green lobby is responsible for GPipe by Anonymous Coward · · Score: 0

    "Now they want us to switch to expensive, inconsistent, polluting GPipe!" -Confusedative

    1. Re: The green lobby is responsible for GPipe by Anonymous Coward · · Score: 0

      Still gonna take forever to train. Infinity divided by whatever is still infinity

    2. Re: The green lobby is responsible for GPipe by Anonymous Coward · · Score: 0

      Behold truth! For the world is hollow, and I have made a pie.

    3. Re: The green lobby is responsible for GPipe by Anonymous Coward · · Score: 0

      Actually X/exp(X) tends to 0 towards infinity for instance.
      Just my 2 cents.

  2. WTF??? Python, LOL by Anonymous Coward · · Score: 0

    Hard pass.

    1. Re:WTF??? Python, LOL by Anonymous Coward · · Score: 0

      A *lot* of big data crunching software is python these days. Interpreted code is still code.

    2. Re: WTF??? Python, LOL by Anonymous Coward · · Score: 0

      Yeah, silly, but should get bindings for real programming languages - tensorflow does.

    3. Re: WTF??? Python, LOL by Anonymous Coward · · Score: 0

      glang is obvious despite 50 shades of glang. I can understand that (spec really). Not java bloat or C++11 boost shitstorms.

  3. which gpipe? by Anonymous Coward · · Score: 0

    python3 -m pip search gpipe
    gpipe (1.0.0) - GridPipe
    torch-gpipe (0.0.0a0) -

    https://pypi.org/project/gpipe/
    https://github.com/east301/gpipe

    Nope!
    Here it is: https://github.com/tensorflow/lingvo/blob/master/lingvo/core/gpipe.py
    I envision Alphabet will sue for the gpipe name.

  4. Blind megalomania by Anonymous Coward · · Score: 0

    Makes you wonder what good will come out of this, and even more what bad things will follow.

    1. Re: Blind megalomania by Anonymous Coward · · Score: 0

      We need to hold these people accountable.

  5. We have to remember what's important here. by JMZero · · Score: 2, Funny

    This software might be very useful for developing computer systems to solve difficult tasks. Sure, whatever... but we can't let that blind us to what's really important here.

    What's truly important here is dissecting exactly what to call these systems. Using the term "neural network" makes sense on pretty much every level, and it would allow us to communicate clearly about what type of algorithms are being used, but that term (and especially "deep neural network") make all my personal bugaboos about AI flare up. These are computers not brains, so there's no neurons, so we can't use that word.

    And yeah, obviously I recognize that, for most people, Google's efforts here are exactly what people mean when they talk about AI - but that makes me angry so nobody should do it. I've decided, for no good reason, that the term AI should only be used when describing an intelligence that works just the same as a human. I have weird quasi-spiritual hangups about all this, which I think you're all obliged to respect.

    PS: Also, just a reminder, Google has accomplished nothing, these systems are useless and aren't improving. All just unimpressive hype. I hate technology and change, please stop. Thanks in advance for never mentioning something like this again.

    --
    Let's not stir that bag of worms...
    1. Re:We have to remember what's important here. by Anonymous Coward · · Score: 0

      " Using the term "neural network" makes sense on pretty much every level " = Full stop. Wrong. We don't understand neural communication any better than we understand AI. It's THE problem.

    2. Re: We have to remember what's important here. by Anonymous Coward · · Score: 0

      Sure, dev-less is next shithead!

    3. Re:We have to remember what's important here. by epine · · Score: 1

      I've decided, for no good reason, that the term AI should only be used when describing an intelligence that works just the same as a human.

      Unfortunately, human intelligence is insufficient (so far as we can tell) to evaluate the predicate "just the same as a human". Which makes things very simple, until we have actual wetware clones that not even Blade Runner can tell apart from the "real" thing.

      I've refused to use the term "AI" in my own notes since the 1980s. I've been using "AC" instead (for artificial cognition). They're basically the same thing, only for AC, Roger Penrose doesn't show up to tell you that you're doing it wrong (I found the central thesis of The Emperor's New Mind extremely annoying, even if the book was entertaining on other levels).

      Beyond "shoo, Roger, shoo", at the end of the day terminological obsessions don't buy much. Prepare to sleep alone.

    4. Re:We have to remember what's important here. by JMZero · · Score: 1

      Fully agree.

      I mean, just because these computer hype things were inspired by biological neurons, unless they function in just the same way then we're insulting God by calling them that.

      It also pisses me off that some arses (many of whom are total jerks to me on the aviation forums) call the side-planks on airplanes "wings", when they don't work anything like bird or insect wings.

      --
      Let's not stir that bag of worms...
  6. i'll be ... by Anonymous Coward · · Score: 0

    laying some gpipe tonight for sure!!

  7. Is that what's really going on in the brain? by GPS+Pilot · · Score: 1

    enabling 318 million parameters on a single accelerator core. Without GPipe, Huang says, a single core can only train up to 82 million model parameters

    Is that what's really going on in the human brain -- hundreds of millions of "model parameters" are getting trained up?

    I doubt it. With this approach, AI researchers are following a road to something completely different from human intelligence. And I'll bet that something will be far more limited than human intelligence.

    --
    That that is is that that that that is not is not.
    1. Re:Is that what's really going on in the brain? by dromgodis · · Score: 1

      With this approach, AI researchers are following a road to something completely different from human intelligence.

      Which is perfectly fine. If you can disregard the debate around the term AI, this road can lead to (and already has) *better* solutions than humans can provide unassisted - for specific problems and quality definitions.

      I don't believe that having one machine, one algorithm to solve all different problems is the right aim at all. And I definitely don't think that a human brain would have been the best target model if that was the aim.

      Making a proper synthetic brain implementation is an interesting endeavour in itself and will provide a lot of useful understanding in medicine, sociology, technology, and a lot of other fields. As a solution to any concrete problems, I think that we can come up with a better solution than the brain for virtually any problem that we can formulate.

    2. Re:Is that what's really going on in the brain? by fbobraga · · Score: 1

      I doubt it.

      I too.