Raw

Лекція 5. Регресійний та кореляційний аналіз

Огляд

У Лекціях 14 ми описували одну випадкову величину: її розподіл, числові характеристики, довірчі інтервали, перевірку гіпотез про середнє чи дисперсію. Але найцікавіші питання аналітики — про зв’язок: чи залежить час відгуку сервера від навантаження, обсяг продажів — від витрат на рекламу, вибір кандидата — від регіону? Тут у гру входять дві змінні одразу.

Ця лекція — про двовимірні дані та про те, як виявити, виміряти й змоделювати залежність між двома змінними. Маршрут визначає природа змінних. Для якісних (категоріальних) змінних ми будуємо таблицю спряженості (contingency table), перевіряємо незалежність критерієм згоди Пірсона (χ2\chi^2) — це прямий нащадок перевірки гіпотез із Лекції 4. Для кількісних змінних ми малюємо діаграму розсіювання, вимірюємо силу лінійного зв’язку коефіцієнтом кореляції rr, будуємо лінійну регресію методом найменших квадратів і оцінюємо її якість коефіцієнтом детермінації R2R^2. Наприкінці — місток до якісних предикторів через фіктивні змінні (dummy variables), який веде до методів класифікації Лекції 6.

Практичний бік. Коефіцієнт кореляції, рівняння регресії та R2R^2 ви обчислите «руками» й програмою в Лабораторній роботі 5.


5.1 Двовимірна статистика

Досі спостереження було одним числом xix_i. Тепер кожне спостереження — це пара (xi,yi)(x_i, y_i): для кожного об’єкта ми фіксуємо дві ознаки одночасно.

Означення (двовимірна вибірка). Двовимірною вибіркою називають набір nn пар (x1,y1),(x2,y2),,(xn,yn)(x_1, y_1), (x_2, y_2), \dots, (x_n, y_n), де кожна пара — це два значення, виміряні на одному й тому самому об’єкті.

Змінні в парі відіграють нерівноправні ролі:

  • Незалежна змінна XX (independent / explanatory variable, регресор, предиктор) — та, що вважається «входом», причиною, керованим фактором.
  • Залежна змінна YY (dependent / response variable, відгук) — та, що ми прагнемо пояснити або спрогнозувати через XX.

Поділ на XX і YY задає дослідник, виходячи зі змісту задачі (витрати на рекламу \to продажі, а не навпаки). Мета двовимірного аналізу — відповісти на три питання: чи є зв’язок між XX і YY; наскільки він сильний; і яка його форма (щоб прогнозувати YY за XX).

За типом змінних розрізняють три ситуації, і кожна має свій інструмент:

XX YY Інструмент
якісна якісна таблиця спряженості, критерій χ2\chi^2
кількісна кількісна кореляція, регресія
якісна кількісна порівняння середніх, фіктивні змінні

5.2 Види залежності

Сказати «YY залежить від XX» можна по-різному, і ці способи утворюють ієрархію від найсильнішого до найслабшого.

  • Повна (статистична) незалежність. Умовний розподіл YY не залежить від значення XX: P(YX=x)=P(Y)P(Y \mid X = x) = P(Y) для всіх xx. Знання XX не дає жодної інформації про YY. Рівносильно — спільна ймовірність факторизується: P(X,Y)=P(X)P(Y)P(X, Y) = P(X)\,P(Y).
  • Незалежність за математичним сподіванням. Слабша умова: від XX не залежить лише середнє YY,   M(YX=x)=M(Y)\;M(Y \mid X = x) = M(Y), хоча розкид чи форма розподілу YY ще можуть змінюватися з XX.
  • Відсутність кореляції (некорельованість). Найслабша умова: дорівнює нулю коваріація Cov(X,Y)=M[(XMX)(YMY)]=0\operatorname{Cov}(X, Y) = M\big[(X - M X)(Y - M Y)\big] = 0, тобто немає лінійного зв’язку.

Властивість (ієрархія). Повна незалежність \Rightarrow незалежність за сподіванням \Rightarrow некорельованість. Зворотні імплікації хибні.

Приклад 5.1 (некорельовані, але залежні). Нехай XX набуває значень 2,1,0,1,2-2, -1, 0, 1, 2 (симетрично), а Y=X2Y = X^2. Тоді YY повністю визначається XX (залежність жорстка), проте вибірковий коефіцієнт кореляції дорівнює рівно нулю: додатні відхилення XX компенсуються від’ємними. Цей приклад — головне застереження курсу: кореляція вловлює лише лінійний зв’язок; r=0r = 0 означає «немає лінійного зв’язку», а не «немає зв’язку взагалі».


5.3 Залежність категоріальних змінних. Таблиця спряженості

Коли обидві змінні якісні, дані зводять у таблицю спряженості — таблицю частот усіх комбінацій категорій.

Означення (таблиця спряженості). Таблицею спряженості розміру r×cr \times c називають таблицю, у клітинці (i,j)(i, j) якої стоїть частота OijO_{ij} — кількість об’єктів, що потрапили одночасно в категорію ii змінної-рядка та категорію jj змінної-стовпця. Суми по рядках і стовпцях називають граничними (маргінальними) підсумками.

Приклад 5.2 (голосування). Досліджують зв’язок між вибором кандидата (рядки) та областю проживання виборця (стовпці). Наведемо фрагмент таблиці — дві області з багатьох; повна вибірка охоплює N=18064N = 18\,064 виборців:

Львівська обл. Харківська обл. \dots Усього
Кандидат 1 548 1126 \dots 13 542
Кандидат 2 676 146 \dots 4 522
Усього 1224 1272 \dots 18 064

Граничні підсумки узгоджені: у Львівській 548+676=1224548 + 676 = 1224, у Харківській 1126+146=12721126 + 146 = 1272; за кандидатами — 13542+4522=18064=N13\,542 + 4\,522 = 18\,064 = N.

Імовірнісна інтерпретація

Поділивши кожну частоту на загальну кількість NN, дістаємо спільні ймовірності P(рядокi,стовпецьj)P(\text{рядок}_i, \text{стовпець}_j), а з граничних підсумків — граничні ймовірності:

Львівська Харківська Усього
Кандидат 1 0.030 0.062 0.750
Кандидат 2 0.037 0.008 0.250
Усього 0.068 0.070 1.000

Наприклад, P(К1,Львів)=548/180640.030P(\text{К1}, \text{Львів}) = 548/18\,064 \approx 0.030, а гранична P(К1)=13542/180640.750P(\text{К1}) = 13\,542/18\,064 \approx 0.750: кандидат 1 набирає загалом близько 75 %.

Умовні ймовірності

Найінформативніший погляд — умовні ймовірності (порівн. Лекцію 1). У розрізі стовпців ділимо клітинку на підсумок стовпця, дістаючи P(кандидатобласть)P(\text{кандидат} \mid \text{область}) — розподіл голосів усередині кожної області:

Львівська Харківська
Кандидат 1 0.448 0.885
Кандидат 2 0.552 0.115
Усього 1.000 1.000

Тут P(К1Львів)=548/12240.448P(\text{К1} \mid \text{Львів}) = 548/1224 \approx 0.448, а P(К1Харків)=1126/12720.885P(\text{К1} \mid \text{Харків}) = 1126/1272 \approx 0.885. У розрізі рядків ділимо на підсумок рядка, дістаючи P(областькандидат)P(\text{область} \mid \text{кандидат}): P(ЛьвівК1)=548/135420.040P(\text{Львів} \mid \text{К1}) = 548/13\,542 \approx 0.040, P(ЛьвівК2)=676/45220.149P(\text{Львів} \mid \text{К2}) = 676/4\,522 \approx 0.149.

Критерій незалежності через ймовірності. Змінні незалежні тоді й лише тоді, коли умовний розподіл збігається з граничним: P(кандидатобласть)=P(кандидат)P(\text{кандидат} \mid \text{область}) = P(\text{кандидат}) для кожної області. У прикладі P(К1Львів)=0.4480.750=P(К1)P(\text{К1} \mid \text{Львів}) = 0.448 \ne 0.750 = P(\text{К1}) — отже, вибір кандидата залежить від області.


5.4 Очікувані частоти за припущення незалежності

Щоб виміряти залежність числом, порівняємо спостережувані частоти OijO_{ij} з тими, які були б очікуваними, якби змінні були незалежні. За незалежності P(рядокi,стовпецьj)=P(рядокi)P(стовпецьj)P(\text{рядок}_i, \text{стовпець}_j) = P(\text{рядок}_i)\,P(\text{стовпець}_j), тож очікувана кількість у клітинці дорівнює NN, помноженому на добуток граничних імовірностей:

Eij=Nсума рядкаiNсума стовпцяjN=(сума рядкаi)(сума стовпцяj)N.E_{ij} = N \cdot \frac{\text{сума рядка}_i}{N} \cdot \frac{\text{сума стовпця}_j}{N} = \frac{(\text{сума рядка}_i)\,(\text{сума стовпця}_j)}{N}.

Приклад 5.3 (очікувані голоси). Кандидат 1 має загальну частку P(К1)=13542/180640.75P(\text{К1}) = 13\,542/18\,064 \approx 0.75. Тоді за незалежності від нього очікувалося б у Львівській області EК1, Львів=135421224/18064918E_{\text{К1, Львів}} = 13\,542 \cdot 1224 / 18\,064 \approx 918 голосів, а фактично O=548O = 548. Аналогічно для двох показаних областей:

Львівська Харківська
Кандидат 1 918 954
Кандидат 2 306 318

Розбіжність велика (548548 проти 918918, 11261126 проти 954954) — сильний сигнал залежності. Щоб перетворити «велику розбіжність» на число з порогом прийняття рішення, застосовують критерій Пірсона.


5.5 Критерій згоди Пірсона (χ2\chi^2)

Критерій χ2\chi^2 формально перевіряє гіпотезу незалежності, порівнюючи всі OijO_{ij} з EijE_{ij} (порівн. алгоритм перевірки гіпотез, Лекція 4).

Означення (статистика Пірсона). Для таблиці спряженості

χ2=i,j(OijEij)2Eij,\chi^2 = \sum_{i, j} \frac{(O_{ij} - E_{ij})^2}{E_{ij}},

де сума береться по всіх клітинках. Число ступенів свободи

df=(r1)(c1).\mathrm{df} = (r - 1)(c - 1).

Логіка. За незалежності OijEijO_{ij} \approx E_{ij}, тож χ2\chi^2 малий; що сильніша залежність, то більший χ2\chi^2. Гіпотези:

  • H0H_0: змінні незалежні;
  • H1H_1: змінні залежні.

Правило рішення. Обравши рівень значущості α\alpha (зазвичай 0.050.05), знаходять критичне значення χкр2=χα,df2\chi^2_{\text{кр}} = \chi^2_{\alpha, \mathrm{df}} (квантиль розподілу χ2\chi^2 з df\mathrm{df} ступенями свободи). Якщо χ2>χкр2\chi^2 > \chi^2_{\text{кр}} — гіпотезу незалежності відхиляють.

Приклад 5.4 (повний розрахунок χ2\chi^2). У групі з 200200 студентів досліджують зв’язок між формою навчання (очна / дистанційна) та результатом іспиту (склав / не склав). Спостережувана таблиця:

Склав Не склав Усього
Очна 80 20 100
Дистанційна 60 40 100
Усього 140 60 200

Теплова карта таблиці спряженості «форма навчання × результат іспиту»: спостережені частоти 80/20/60/40 та очікувані за незалежності E=70/30/70/30; хі-квадрат = 9.52

Крок 1. Очікувані частоти Eij=(рядокi)(стовпецьj)/NE_{ij} = (\text{рядок}_i)(\text{стовпець}_j)/N:

E11=100140200=70,E12=10060200=30,E21=70,E22=30.E_{11} = \frac{100 \cdot 140}{200} = 70, \quad E_{12} = \frac{100 \cdot 60}{200} = 30, \quad E_{21} = 70, \quad E_{22} = 30.

Крок 2. Внески клітинок (OE)2/E(O - E)^2 / E:

(8070)270=10070=1.4286,(2030)230=10030=3.3333,\frac{(80 - 70)^2}{70} = \frac{100}{70} = 1.4286, \qquad \frac{(20 - 30)^2}{30} = \frac{100}{30} = 3.3333,

(6070)270=1.4286,(4030)230=3.3333.\frac{(60 - 70)^2}{70} = 1.4286, \qquad \frac{(40 - 30)^2}{30} = 3.3333.

Крок 3. Статистика й рішення.

χ2=1.4286+3.3333+1.4286+3.3333=9.524,df=(21)(21)=1.\chi^2 = 1.4286 + 3.3333 + 1.4286 + 3.3333 = 9.524, \qquad \mathrm{df} = (2 - 1)(2 - 1) = 1.

Критичне значення χ0.05,12=3.841\chi^2_{0.05,\,1} = 3.841. Оскільки 9.524>3.8419.524 > 3.841, гіпотезу незалежності відхиляємо: форма навчання та результат іспиту пов’язані (на очній частка тих, хто склав, вища).

Умови застосовності. Критерій наближений і надійний, коли очікувані частоти достатньо великі — практичне правило: Eij5E_{ij} \ge 5 у всіх клітинках. Для таблиць 2×22 \times 2 програмні пакети часто застосовують поправку Єйтса (неперервності), яка трохи зменшує χ2\chi^2; щоб відтворити «формульне» значення (OE)2/E\sum (O - E)^2 / E, у scipy.stats.chi2_contingency вимикають поправку (correction=False).

Типова помилка (частоти, а не частки). χ2\chi^2 рахують лише за абсолютними частотами (кількостями). Підставляти у формулу відсотки чи ймовірності не можна — результат втратить зв’язок з обсягом вибірки NN і буде беззмістовним.


5.6 Залежності в кількісних даних. Діаграма розсіювання

Коли обидві змінні кількісні, перший крок аналізу — діаграма розсіювання (scatter plot): кожну пару (xi,yi)(x_i, y_i) зображують точкою на площині (XX — по горизонталі, YY — по вертикалі). Оком з діаграми зчитують:

  • напрям — зростає YY з XX (додатній зв’язок) чи спадає (від’ємний);
  • силу — наскільки щільно точки лягають уздовж уявної лінії;
  • форму — лінійна залежність чи криволінійна;
  • викиди — окремі точки, що різко вибиваються із загальної картини.

Діаграма розсіювання — обов’язкова передумова кореляційно-регресійного аналізу: вона показує, чи взагалі доречна лінійна модель (згадайте Приклад 5.1, де зв’язок є, але він не лінійний).


5.7 Вибірковий коефіцієнт кореляції

Діаграма дає якісне враження; коефіцієнт кореляції перетворює його на число.

Чотири діаграми розсіювання з різними значеннями коефіцієнта кореляції: сильна додатна r≈0.96, помірна додатна r≈0.48, відсутність зв'язку r≈0.01 та сильна від'ємна r≈−0.88

Означення (вибірковий коефіцієнт кореляції Пірсона).

r=i=1n(xixˉ)(yiyˉ)i=1n(xixˉ)2  i=1n(yiyˉ)2. r = \frac{\sum_{i=1}^{n} (x_i - \bar{x})(y_i - \bar{y})} {\sqrt{\sum_{i=1}^{n} (x_i - \bar{x})^2 \; \sum_{i=1}^{n} (y_i - \bar{y})^2}}.

Чисельник — це (з точністю до множника 1/n1/n) коваріація, а знаменник нормує її на розкиди обох змінних, тож rr безрозмірний.

Властивості. 1r1-1 \le r \le 1; знак показує напрям зв’язку; r=1|r| = 1 — точний лінійний зв’язок (усі точки на одній прямій); r=0r = 0 — лінійного зв’язку немає. Орієнтовна шкала сили:

| r|r| | Сила лінійного зв’язку | |:–:|—| | 0.00.00.30.3 | дуже слабкий / відсутній | | 0.30.30.50.5 | слабкий | | 0.50.50.70.7 | помірний | | 0.70.70.90.9 | сильний | | 0.90.91.01.0 | дуже сильний |

Для ручних обчислень зручніша алгебраїчно рівносильна форма через суми (без попереднього віднімання середніх):

r=nxiyixiyi[nxi2(xi)2][nyi2(yi)2].r = \frac{n \sum x_i y_i - \sum x_i \sum y_i} {\sqrt{\big[n \sum x_i^2 - (\sum x_i)^2\big]\big[n \sum y_i^2 - (\sum y_i)^2\big]}}.

Значущість кореляції. Навіть за відсутності зв’язку в генеральній сукупності вибірковий rr рідко дорівнює точно нулю. Значущість перевіряють статистикою

t=rn21r2,df=n2,t = r\,\sqrt{\frac{n - 2}{1 - r^2}}, \qquad \mathrm{df} = n - 2,

порівнюючи t|t| з критичним tα/2,n2t_{\alpha/2,\,n-2} (розподіл Стьюдента, Лекція 2). Це та сама логіка перевірки гіпотез, що й у Лекції 4, з H0 ⁣:r=0H_0\!: r = 0.


5.8 Основа регресійного аналізу. Метод найменших квадратів

Кореляція вимірює силу зв’язку; регресія задає його форму — рівняння, за яким прогнозують YY.

Означення (проста лінійна регресія). Лінійною регресійною моделлю називають y^=b0+b1x\hat{y} = b_0 + b_1 x, де b1b_1коефіцієнт нахилу (на скільки в середньому змінюється YY зі зростанням XX на одиницю), b0b_0вільний член (значення y^\hat{y} при x=0x = 0). Різницю ei=yiy^ie_i = y_i - \hat{y}_i між фактом і прогнозом називають залишком (residual).

Метод найменших квадратів (ordinary least squares, OLS) обирає пряму, що мінімізує суму квадратів залишків ei2=(yib0b1xi)2\sum e_i^2 = \sum (y_i - b_0 - b_1 x_i)^2. Прирівнявши частинні похідні за b0,b1b_0, b_1 до нуля, дістають єдиний розв’язок:

b1=(xixˉ)(yiyˉ)(xixˉ)2,b0=yˉb1xˉ.b_1 = \frac{\sum (x_i - \bar{x})(y_i - \bar{y})}{\sum (x_i - \bar{x})^2}, \qquad b_0 = \bar{y} - b_1 \bar{x}.

Формула для b0b_0 означає, що пряма регресії завжди проходить через центр даних (xˉ,yˉ)(\bar{x}, \bar{y}). Нахил b1b_1 і кореляція rr пов’язані: b1=rsy/sxb_1 = r \cdot s_y / s_x — тому знаки b1b_1 і rr завжди збігаються.

Усі величини обчислюють за один прохід по даних, накопичуючи п’ять сум:

Вхід: пари (x_1, y_1), …, (x_n, y_n)
n ← 0; Sx ← 0; Sy ← 0; Sxy ← 0; Sxx ← 0; Syy ← 0
для кожної пари (x, y):
    n ← n + 1
    Sx ← Sx + x;   Sy ← Sy + y
    Sxy ← Sxy + x*y
    Sxx ← Sxx + x*x;  Syy ← Syy + y*y
Dxy ← n*Sxy − Sx*Sy
Dxx ← n*Sxx − Sx*Sx
Dyy ← n*Syy − Sy*Sy
r  ← Dxy / sqrt(Dxx * Dyy)
b1 ← Dxy / Dxx
b0 ← (Sy − b1*Sx) / n
R2 ← r*r                      // для простої лінійної регресії
Вихід: r, b0, b1, R2

Приклад 5.5 (побудова регресії). Досліджують зв’язок між кількістю годин підготовки XX та балом за тест YY для n=6n = 6 студентів. Складемо розрахункову таблицю:

xix_i yiy_i xiyix_i y_i xi2x_i^2 yi2y_i^2
2 35 70 4 1225
4 45 180 16 2025
5 48 240 25 2304
6 58 348 36 3364
8 72 576 64 5184
9 78 702 81 6084
\sum
34 336 2116 226 20186

Тут n=6n = 6, xˉ=34/65.667\bar{x} = 34/6 \approx 5.667, yˉ=336/6=56\bar{y} = 336/6 = 56. Проміжні величини:

nxyxy=6211634336=1269611424=1272,n\sum x y - \sum x \sum y = 6 \cdot 2116 - 34 \cdot 336 = 12\,696 - 11\,424 = 1272,

nx2(x)2=6226342=13561156=200,n\sum x^2 - (\sum x)^2 = 6 \cdot 226 - 34^2 = 1356 - 1156 = 200,

ny2(y)2=6201863362=121116112896=8220.n\sum y^2 - (\sum y)^2 = 6 \cdot 20\,186 - 336^2 = 121\,116 - 112\,896 = 8220.

Тоді

b1=1272200=6.36,b0=566.365.667=19.96,b_1 = \frac{1272}{200} = 6.36, \qquad b_0 = 56 - 6.36 \cdot 5.667 = 19.96,

r=12722008220=12721282.2=0.9921.r = \frac{1272}{\sqrt{200 \cdot 8220}} = \frac{1272}{1282.2} = 0.9921.

Рівняння регресії: y^=19.96+6.36x\hat{y} = 19.96 + 6.36\,x. Кожна додаткова година підготовки додає в середньому 6.4\approx 6.4 бала. Прогноз для x=7x = 7 годин: y^=19.96+6.367=64.5\hat{y} = 19.96 + 6.36 \cdot 7 = 64.5 бала.

Діаграма розсіювання балів за тест від годин підготовки з лінією регресії найменших квадратів ŷ = 19.96 + 6.36·x, позначеним центром даних та R^2 = 0.984

Значущість. t=0.99214/(10.99212)=15.77t = 0.9921\sqrt{4 / (1 - 0.9921^2)} = 15.77; критичне t0.025,4=2.776t_{0.025,\,4} = 2.776. Оскільки 15.77>2.77615.77 > 2.776, кореляція значуща.

Типова помилка (екстраполяція). Рівняння y^=b0+b1x\hat{y} = b_0 + b_1 x надійне лише в межах діапазону спостережуваних xx. Підставляти значення далеко за межами вибірки (напр., x=0x = 0, коли всі дані мають x2x \ge 2) — ризиковано: модель там не перевірена, а b0b_0 може навіть не мати змістовної інтерпретації.


5.9 Коефіцієнт детермінації та дисперсійний аналіз

Наскільки добре пряма описує дані? Відповідь дає розклад суми квадратів. Загальний розкид YY навколо середнього розпадається на дві частини — пояснену регресією та залишкову:

(yiyˉ)2SST — повна=(y^iyˉ)2SSR — пояснена+(yiy^i)2SSE — залишкова.\underbrace{\sum (y_i - \bar{y})^2}_{\text{SST — повна}} = \underbrace{\sum (\hat{y}_i - \bar{y})^2}_{\text{SSR — пояснена}} + \underbrace{\sum (y_i - \hat{y}_i)^2}_{\text{SSE — залишкова}}.

  • SST (total sum of squares) — уся варіація YY;
  • SSR (regression) — варіація, яку пояснює модель;
  • SSE (error) — варіація залишків, яку модель не пояснює.

Геометричний розклад відхилення точки від середнього на пояснену частину (від середнього до лінії регресії) та залишкову (від лінії до точки) — основа розкладу SST = SSR + SSE

Означення (коефіцієнт детермінації). Частка поясненої варіації

R2=SSRSST=1SSESST,0R21.R^2 = \frac{\text{SSR}}{\text{SST}} = 1 - \frac{\text{SSE}}{\text{SST}}, \qquad 0 \le R^2 \le 1.

Для простої лінійної регресії R2=r2R^2 = r^2.

Приклад 5.6 (продовження 5.5). Для тієї самої моделі SST=(ny2(y)2)/n=8220/6=1370\text{SST} = (n\sum y^2 - (\sum y)^2)/n = 8220/6 = 1370. Обчисливши прогнози y^i\hat{y}_i та залишки, дістаємо SSR=1348.32\text{SSR} = 1348.32 і SSE=21.68\text{SSE} = 21.68 (перевірка: 1348.32+21.68=1370=SST1348.32 + 21.68 = 1370 = \text{SST}). Тоді

R2=1348.321370=0.9842=r2=0.99212.R^2 = \frac{1348.32}{1370} = 0.9842 = r^2 = 0.9921^2.

Модель пояснює 98.4 % розкиду балів — години підготовки майже повністю визначають результат. Так R2R^2 стає мірою якості підгонки: що ближчий до 1, то щільніше точки лягають на пряму.

Графік залишків регресії: вертикальні відрізки від нуля до кожного залишку; залишки невеликі й безсистемно розкидані навколо нуля, що свідчить про адекватність лінійної моделі


5.10 Залежність кількісного від якісного. Фіктивні змінні

Що робити, коли предиктор якісний (стать, регіон, тип пристрою)? Категорію не можна прямо помножити на b1b_1. Розв’язок — фіктивна (індикаторна) змінна.

Означення (фіктивна змінна). Фіктивна змінна (dummy variable) DD кодує наявність категорії числами 00 та 11: D=1D = 1, якщо об’єкт належить категорії, і D=0D = 0 інакше.

У моделі y^=b0+b1D\hat{y} = b_0 + b_1 D коефіцієнт b0b_0 дорівнює середньому YY у базовій групі (D=0D = 0), а b1b_1різниці середніх між групами.

Дві групи точок при значеннях фіктивної змінної d = 0 та d = 1; пряма з'єднує середні груп: вільний член дорівнює середньому базової групи, а нахил — різниці середніх між групами

Приклад 5.7. Порівнюють зарплату у двох відділах. Кодуємо відділ як DD (00 — відділ A, 11 — відділ B). Нехай середні зарплати 4040 і 5050 тис. Регресія YY на DD дає b0=40b_0 = 40 (середнє в A) та b1=10b_1 = 10 (наскільки B у середньому вище), тобто y^=40+10D\hat{y} = 40 + 10 D.

Для якісної змінної з kk категоріями беруть k1k - 1 фіктивних змінних (одна категорія лишається базовою). Брати всі kk не можна:

Типова помилка (пастка фіктивних змінних). Якщо ввести фіктивні змінні для всіх kk категорій разом із вільним членом, їхня сума тотожно дорівнює 11 — виникає повна колінеарність (dummy variable trap), і коефіцієнти визначити неможливо. Завжди лишайте одну категорію базовою.

Фіктивні змінні поєднують кількісні та якісні предиктори в одній моделі й ведуть до множинної регресії та методів класифікації Лекції 6.


Застосування в аналітиці даних

  • Прогнозування. Лінійна регресія — базовий інструмент прогнозу відгуку за предиктором (продажі за рекламою, навантаження за кількістю користувачів).
  • Виявлення зв’язків. Таблиці спряженості та χ2\chi^2 показують, чи пов’язані категоріальні ознаки (канал залучення й конверсія, регіон і вибір продукту).
  • Оцінка якості моделі. R2R^2 — стандартна метрика частки поясненої варіації; розклад SST == SSR ++ SSE лежить в основі дисперсійного аналізу (ANOVA).
  • Обережність з причинністю. Кореляція та значуща регресія свідчать про зв’язок, а не про причину: за ним може стояти прихований третій фактор.
  • Підготовка ознак. Кодування категорій фіктивними змінними — обов’язковий крок перед подачею даних у більшість моделей машинного навчання.

Підсумок

  • Двовимірна вибірка — пари (xi,yi)(x_i, y_i); XX — незалежна змінна, YY — залежна. Види незалежності за спаданням сили: повна \Rightarrow за сподіванням \Rightarrow некорельованість; зворотне не так (Y=X2Y = X^2 дає r=0r = 0).
  • Для якісних змінних будують таблицю спряженості; спільні, граничні та умовні ймовірності описують зв’язок. Незалежність     \iff умовний розподіл дорівнює граничному.
  • Очікувані частоти за незалежності: Eij=(рядокi)(стовпецьj)/NE_{ij} = (\text{рядок}_i)(\text{стовпець}_j)/N.
  • Критерій Пірсона: χ2=(OE)2/E\chi^2 = \sum (O - E)^2 / E, df=(r1)(c1)\mathrm{df} = (r-1)(c-1); порівнюють з χα,df2\chi^2_{\alpha,\,\mathrm{df}}.
  • Для кількісних змінних: діаграма розсіювання, коефіцієнт кореляції r[1,1]r \in [-1, 1] (лише лінійний зв’язок).
  • Метод найменших квадратів: b1=(xxˉ)(yyˉ)(xxˉ)2b_1 = \frac{\sum (x - \bar{x})(y - \bar{y})}{\sum (x - \bar{x})^2}, b0=yˉb1xˉb_0 = \bar{y} - b_1 \bar{x}; пряма проходить через (xˉ,yˉ)(\bar{x}, \bar{y}).
  • Якість: SST == SSR ++ SSE; R2=SSR/SSTR^2 = \text{SSR}/\text{SST}; для простої лінійної регресії R2=r2R^2 = r^2.
  • Фіктивні змінні кодують якісний предиктор (0/10/1); для kk категорій — k1k - 1 змінних (уникайте пастки колінеарності).

Вправи

Для розігріву

  1. Поясніть різницю між незалежною (XX) та залежною (YY) змінними на власному прикладі. Хто задає цей поділ?
  2. Чому з r=0r = 0 не випливає незалежність змінних? Наведіть приклад.
  3. Для таблиці спряженості 3×43 \times 4 обчисліть число ступенів свободи критерію χ2\chi^2.
  4. Кореляція між зростом і вагою дорівнює r=0.8r = 0.8. Що можна сказати про силу та напрям зв’язку? Скільки відсотків варіації ваги пояснює лінійна модель?

Стандартні

  1. У таблиці спряженості 2×22 \times 2 маємо частоти 30102040\begin{smallmatrix} 30 & 10 \\ 20 & 40 \end{smallmatrix}. Обчисліть очікувані частоти, статистику χ2\chi^2 та порівняйте з χ0.05,12=3.841\chi^2_{0.05,\,1} = 3.841. Зробіть висновок про незалежність.
  2. Для вибірки (1,3),(2,4),(3,7),(4,8),(5,8)(1, 3), (2, 4), (3, 7), (4, 8), (5, 8) обчисліть rr, рівняння регресії y^=b0+b1x\hat{y} = b_0 + b_1 x та R2R^2. Спрогнозуйте yy при x=6x = 6.
  3. Доведіть, що пряма найменших квадратів проходить через точку (xˉ,yˉ)(\bar{x}, \bar{y}), виходячи з формули b0=yˉb1xˉb_0 = \bar{y} - b_1 \bar{x}.
  4. Поясніть зміст SST, SSR і SSE. Що означає R2=0.4R^2 = 0.4 для якості моделі?

Підвищеної складності

  1. Виведіть формули методу найменших квадратів, мінімізуючи S(b0,b1)=(yib0b1xi)2S(b_0, b_1) = \sum (y_i - b_0 - b_1 x_i)^2 (прирівняйте частинні похідні S/b0\partial S / \partial b_0 та S/b1\partial S / \partial b_1 до нуля).
  2. Доведіть тотожність R2=r2R^2 = r^2 для простої лінійної регресії, скориставшись зв’язком b1=rsy/sxb_1 = r\, s_y / s_x та означеннями SSR і SST.
  3. Якісна змінна має 44 категорії. Скільки фіктивних змінних потрібно й чому не можна взяти всі чотири разом із вільним членом? Опишіть, як інтерпретувати коефіцієнти отриманої моделі.
  4. Поясніть, чому статистично значуща кореляція не доводить причинного зв’язку. Наведіть приклад удаваної кореляції (spurious correlation) через прихований третій фактор.

Lectures/DA-L05.md · 36.4 KB · updated 2026-08-05 09:18