Scalding: Как уменьшить вычисления в памяти в списках?

Question

С помощью Scalding я пытаюсь найти расстояние редактирования между парами похожих строк. Всего у меня есть 10 000 000 строк в файле CSV. Для сокращения вычислений я использую следующий алгоритм:

1. Разделите все строки в группы, используя первые три символа в качестве ключа
2. Генерация комбинаций из двух строк в каждой группе
3. Найти расстояние редактирования для каждой пары строк в каждой группе (см. Код ниже)

Когда я запускаю этот алгоритм на HDFS, он работает с 1 000 000 строк. С 10 000 000 строк менеджер узлов жалуется на попытки моей "карты" выделить больше физической памяти, чем у узла. Я понимаю, что это условие возникает, когда создается огромное количество комбинаций в .groupBy('key) { _.mapList ...} код. Конечно, этот алгоритм на самом деле не масштабируется.

Посоветуйте, пожалуйста, другие подходы для сокращения вычислений для этой задачи.

import cascading.tuple.Fields
import com.twitter.scalding._

class Proto1(args: Args) extends Job(args) {
  val outputStr = "tmp/out.txt"
  val output = TextLine(outputStr)

  val wordsList = List(
    ("aaaa"),
    ("aaaa"),
    ("aaaad"),
    ("aaaab"),
    ("aaabb"),
    ("aabbcc"),
    ("aaabccdd"),
    ("aaabbccdde"),
    ("aaabbddd"),
    ("bbbb"),
    ("bbbb"),
    ("bbbaaa"),
    ("bbaaabb"),
    ("bbbcccc"),
    ("bbbddde"),
    ("bbbdddd"),
    ("bbbdddf"),
    ("bbbdddd"),
    ("ccccc"),
    ("cccaaa"),
    ("ccccaabbb"),
    ("ccbbbddd"),
    ("cdddeee"),
    ("ddd"),
    ("ddd")
    )

  val orderedPipe =
    IterableSource[(String)](wordsList, ('word))
      .map('word -> 'key) { word: String => word.take(3) }
      .debug
      //.groupBy('key) { _.toList[String]('word -> 'list) }
      //.debug
      .groupBy('key) { _.mapList[String, List[(String, String, Int)]]('word -> 'list)(editDistances) }
      .filterNot('key, 'list) { fields: (String, List[(String, String, Int)]) =>
        val (key, list) = fields
        list.isEmpty
      }
      .flatMapTo('list -> ('str1, 'str2, 'dist)) {list:List[(String, String, Int)] => list  }
      .debug
      //.write(output)
      .write(Csv(outputStr))

  def editDistances(list: List[String]): List[(String, String, Int)] = {

    val resultList = list.combinations(2).toList.map(x => (x(0), x(1), editDistance(x(0), x(1))))
        //println(resultList+"\n")
    val result = resultList.filter(x => x._3 <= 1)
    result
  }

  def editDistance(a: String, b: String): Int = {

    import scala.math.min

    def min3(x: Int, y: Int, z: Int) = min(min(x, y), z)

    val (m, n) = (a.length, b.length)

    val matrix = Array.fill(m + 1, n + 1)(0)

    for (i <- 0 to m; j <- 0 to n) {

      matrix(i)(j) = if (i == 0) j
      else if (j == 0) i
      else if (a(i - 1) == b(j - 1)) matrix(i - 1)(j - 1)
      else min3(
        matrix(i - 1)(j) + 1,
        matrix(i)(j - 1) + 1,
        matrix(i - 1)(j - 1) + 1)
    }

    matrix(m)(n)
  }

}

Есть идеи?