HardAssignments

A3 — Optimizer Comparison on a Large Embedding

Optimization · Sinkhorn

Hard

Prerequisites (this site)

Readings

Problem

Compare Adam-style, row-normalized and row+column-normalized (Sinkhorn) updates on a large embedding matrix.

Input

Log update RMS statistics per scheme and the downstream metric.

Output

A table of RMS + metric per scheme and a short note on when Sinkhorn wins.

Requirements

Constraints

Hints

  1. Fix all variables except the one you are sweeping.
  2. Instrument cost (FLOPs/bytes/time) before tuning quality.
  3. Write the honest limitation before writing the conclusion.

Assignment for CS/AI 684.