Вероятность и статистика · Линейная регрессия

Линейная регрессия

Простая линейная регрессия подбирает прямую ŷ = a + bx, для которой сумма квадратов вертикальных остатков наименьшая. Здесь можно сдвигать и поворачивать свою прямую в поисках этого минимума и сравнивать подобранную прямую с облаком, которое она описывает.

Линейная регрессия на этом рисунке

Простая линейная регрессия предсказывает yy по xx с помощью прямой наименьших квадратов y^=a+bx\hat y = a + bx — прямой с наименьшей суммой квадратов остатков. Остаток точки e=y−y^e = y - \hat y показывает, насколько она выше прямой; ниже прямой он отрицателен. Когда точек мало, каждому остатку соответствует квадрат, и сумма становится видимой площадью. Можно сдвигать и поворачивать свою прямую в поисках минимума и сравнивать подобранную прямую со смоделированным облаком из 1000 точек.

Почему прямая проходит через центр (xˉ,yˉ)(\bar x, \bar y)? Запишем любую прямую как y=h+b(x−xˉ)y = h + b(x - \bar x), где hh — её высота при xˉ\bar x. Сумма её остатков равна n(yˉ−h)n(\bar y - h), а сумма квадратов делится на две части:

Сумма квадратов для любой прямой: часть от высоты и часть от наклона
SSE=n(yˉ−h)2+P(b),P(b)=Syy−2b Sxy+b2Sxx\mathrm{SSE} = n(\bar y - h)^2 + P(b), \qquad P(b) = S_{yy} - 2b\,S_{xy} + b^2 S_{xx}

Здесь SxxS_{xx}, SyyS_{yy} и SxyS_{xy} — суммы квадратов и произведений отклонений от средних. Первая часть зависит только от высоты, вторая — только от наклона. Поэтому при любом наклоне лучшая высота — yˉ\bar y: наилучшая прямая проходит через центр, а сумма её остатков равна 0. При повороте вокруг центра сумма следует параболе P(b)P(b). Если значения xx не все одинаковы, её единственный минимум достигается при b=Sxy/Sxxb = S_{xy}/S_{xx}.

Что измеряет R2R^2? Горизонтальная прямая y=yˉy = \bar y соответствует точке параболы при наклоне 0, а сумма её квадратов равна общей вариации SST=Syy\mathrm{SST} = S_{yy}. Подобранная прямая оставляет SSE\mathrm{SSE}; если значения yy меняются, то R2=1−SSE/SSTR^2 = 1 - \mathrm{SSE}/\mathrm{SST} — доля вариации yy, которую она объясняет. Когда меняются обе координаты, R2=r2R^2 = r^2, поэтому корреляция 0.5 объясняет лишь четверть. «Объясняет» здесь означает «описывает подгонкой», а не «вызывает».

Почему прямая более пологая, чем облако? В стандартизованных единицах её предсказания равны z^y=r zx\hat z_y = r\,z_x. Поэтому при ∣r∣<1|r| < 1 для точки вдали от xˉ\bar x предсказание находится на меньшем числе стандартных отклонений от yˉ\bar y, чем сама точка от xˉ\bar x. В овальном облаке, подобном смоделированному, прямая стандартных отклонений с наклоном ±sy/sx\pm s_y/s_x идёт вдоль видимого направления облака, а прямая регрессии более пологая. Ничто не тянет данные назад: эффект верен для любых таких данных. Предсказание xx по yy показывает его в обратном направлении и даёт вторую прямую через центр, более крутую на графике. В таком облаке средние yy в узких полосах xx показывают этот эффект: они близки к прямой регрессии, но не лежат точно на ней. Название происходит от исследования роста, проведённого Фрэнсисом Гальтоном в 1886 году: дети очень высоких или очень низких родителей в среднем были ближе к среднему росту.

Когда прямая может ввести в заблуждение? Если поднять точку на 1, её собственное предсказанное значение вырастет на её рычаг h=1/n+(x−xˉ)2/Sxxh = 1/n + (x - \bar x)^2/S_{xx}, который увеличивается с расстоянием от xˉ\bar x. У отдельной точки, далеко отстоящей по xx, рычаг может быть близок к 1: прямая следует за ней. Даже если точка противоречит всем остальным, её собственный остаток может выглядеть небольшим; даже в большом облаке такая точка может повернуть прямую. И rr может быть близко к 1, хотя остатки положительны по обоим краям и отрицательны посередине: получается кривая, которой не следует никакая прямая.

На что обратить внимание

  • При любом масштабе осей квадраты на экране остаются квадратами, поэтому их площади соотносятся как e2e^2.

Управление

  • Нажатие на точку облака показывает её остаток и рычаг; нажатие на пустое место добавляет вашу точку.
  • На клавиатуре ↑ ↓ перемещают центр вращения вашей прямой, а рукоятка поворота вращается на 1° за шаг. Если шаг пересекает центр или горизонтальное положение, движение останавливается там; при видимых ответах то же происходит при пересечении наклона наименьших квадратов.

Связанные понятия