Upload trained artifacts for CheXVision-ResNet_best

CheXVision-ResNet_best.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a1e6c63b65cb596fc9a8e485b7f1bb40dae3bfe856ef8c9574b0330217e18a5f
-size 257688938
+oid sha256:eebff599d39394890c23208e15bc089a56d888509ecd687034f735d0f9e05571
+size 257689002

CheXVision-ResNet_history.json

@@ -1,413 +1,388 @@
 {
   "train_combined_loss": [
-    1.7193066400243064,
-    1.6319774799675641,
-    1.5921577036649808,
-    1.5715855851885623,
-    1.5526096972227585,
-    1.5375647344548145,
-    1.5276719710598923,
-    1.51471095159497,
-    1.5020430373470461,
-    1.4911687616984963,
-    1.4827118652689686,
-    1.4645557582843094,
-    1.4599297595014327,
-    1.4396859422833597,
-    1.435539446407329,
-    1.4271332907735397,
-    1.4179421217166648,
-    1.403216678831095,
-    1.4028082510267657,
-    1.3912968738283542,
-    1.3905528351132577,
-    1.368464354100069,
-    1.3732676605098209,
-    1.3612577147693188,
-    1.3580954211825564,
-    1.349657351700654,
-    1.3462725321582267,
-    1.3339615156037132,
-    1.328426195428564,
-    1.3237001741328687,
-    1.321514273648581,
-    1.319026428947793,
-    1.309424513644452,
-    1.300233209015749,
-    1.2933348074378075,
-    1.2864114295081326,
-    1.2829309904188908,
-    1.2802288567232312,
-    1.2804806562393574,
-    1.2689745398784789,
-    1.2584112556830849,
-    1.2571758327300204,
-    1.2469394986728297,
-    1.2496043610484766,
-    1.2369848253739677,
-    1.2362593363482883,
-    1.2238024343561822,
-    1.2261232994695546,
-    1.2163730117939122,
-    1.2134837068443078,
-    1.203720935823294,
-    1.1993581782514364,
-    1.1956696627469743,
-    1.1901293286887846,
-    1.1828468518517377,
-    1.1760831997685126,
-    1.1743592221471781,
-    1.1676305691189193,
-    1.16071797906741,
-    1.1500692465875693,
-    1.1462531019587667,
-    1.1401782409304422,
-    1.1347792832196908,
-    1.1247655882373617,
-    1.1250666635027664,
-    1.1117426747907906,
-    1.1028407062229975,
-    1.1057177975278534,
-    1.098347607894564,
-    1.0851252087814964,
-    1.0856234678294652,
-    1.0743625631672855,
-    1.0731956925996757,
-    1.0702588654317766,
-    1.0624129944861205,
-    1.0540558592334945,
-    1.0523786074957693,
-    1.0476193412386299,
-    1.043437443798959,
-    1.0361742765922062
+    1.7124134723719688,
+    1.6563327389704188,
+    1.6192178167574733,
+    1.6065538301949283,
+    1.5934712331857854,
+    1.5737339528240253,
+    1.5592873922382071,
+    1.5484244657245259,
+    1.5384113620743087,
+    1.5241506459640335,
+    1.5128174654407918,
+    1.5028224598814723,
+    1.480468608922832,
+    1.4675930849072676,
+    1.4522311548043412,
+    1.4432593131292852,
+    1.431549858463548,
+    1.4213039513256558,
+    1.418550699499799,
+    1.4040459446372822,
+    1.3963383797793507,
+    1.3931131281644318,
+    1.3775085430579685,
+    1.3759366764072714,
+    1.362007566436176,
+    1.3652394998069908,
+    1.3587400428182568,
+    1.3484214210224064,
+    1.3434770855767135,
+    1.332557792918944,
+    1.3278926902824124,
+    1.3267917664060596,
+    1.314020259485277,
+    1.3158454730093168,
+    1.3049792313289554,
+    1.2918729222382204,
+    1.294607077682521,
+    1.2838949585518642,
+    1.283808339243854,
+    1.2675277957111992,
+    1.270660751010499,
+    1.2610519348052363,
+    1.261400941574746,
+    1.256242699845456,
+    1.2451352818117174,
+    1.2347431874488015,
+    1.2273550647410953,
+    1.2298510473997786,
+    1.2254975177794833,
+    1.2154513573822663,
+    1.2096253557181718,
+    1.203949769062789,
+    1.1943704110360138,
+    1.1908197847099515,
+    1.1870963918902244,
+    1.1816325377192487,
+    1.1757468714755144,
+    1.1663964001278981,
+    1.1685066475028734,
+    1.156439424239881,
+    1.146091536159844,
+    1.1453223183141117,
+    1.135295318801354,
+    1.1266746980001392,
+    1.1235529569597675,
+    1.1187340588560835,
+    1.1156733670246421,
+    1.1034497409719584,
+    1.0999683681067116,
+    1.0941586495363078,
+    1.0909428912223247,
+    1.0838057912471075,
+    1.0761769304058302,
+    1.0813932184000388,
+    1.0619072876698643
   ],
   "val_multilabel_loss": [
-    1.3587770308576412,
-    1.2979212917053877,
-    1.2737988195757368,
-    1.2421057458911369,
-    1.2411232156984842,
-    1.2454890123943785,
-    1.2187369696017523,
-    1.208064973910353,
-    1.1996844536555347,
-    1.1994789546327804,
-    1.187707005819278,
-    1.1633295326757787,
-    1.1674125660488854,
-    1.1689632831891972,
-    1.1764915106234266,
-    1.182024956416728,
-    1.171085882320333,
-    1.1566256758437228,
-    1.1557088037257763,
-    1.1606980737910342,
-    1.1157113055684673,
-    1.1234864555855295,
-    1.1408849361894735,
-    1.1247840421413309,
-    1.1282216651893373,
-    1.126798038273605,
-    1.1151826614096982,
-    1.1377070539478045,
-    1.1349695773044628,
-    1.1394831799526712,
-    1.1823002585946625,
-    1.1979565246781307,
-    1.1515015513816875,
-    1.1616137074445612,
-    1.2111144113674093,
-    1.1897162462570774,
-    1.2482671467448347,
-    1.1460777402146538,
-    1.2085615902471898,
-    1.1396334383692315,
-    1.2153222138868338,
-    1.1863896009414943,
-    1.2663917332442838,
-    1.2043939986780507,
-    1.2805147536877375,
-    1.2789906082091047,
-    1.2600323416626276,
-    1.3514712900479338,
-    1.329880796150485,
-    1.3387917110613032,
-    1.3195175874922702,
-    1.4069586119322635,
-    1.3508582265852993,
-    1.322483478364215,
-    1.40722329714405,
-    1.4136808580204623,
-    1.418752995333565,
-    1.3904325611778159,
-    1.538996912149796,
-    1.5857611665529991,
-    1.4956775041880892,
-    1.5053675744070936,
-    1.540437321831931,
-    1.5240953742568173,
-    1.5948855673977689,
-    1.618571667973675,
-    1.602688939380112,
-    1.6135349103541516,
-    1.6695203668257195,
-    1.6448774825948387,
-    1.7579388151417916,
-    1.6869916798018698,
-    1.6719623963112262,
-    1.6472535990734598,
-    1.8512346063976857,
-    1.779941865090114,
-    1.8332922462540775,
-    1.859218388517846,
-    1.8297978248169173,
-    1.9444677036867213
+    1.432891663383035,
+    1.3367359573099793,
+    1.3152825094404674,
+    1.3294410660487264,
+    1.294724655251543,
+    1.2650490028350627,
+    1.2481751419415994,
+    1.2237949782058972,
+    1.2293956086081284,
+    1.2448742656814618,
+    1.2137248326583403,
+    1.1800265104329886,
+    1.1766311911975635,
+    1.1870563145445174,
+    1.15466878293943,
+    1.1560479532770749,
+    1.1556474978850335,
+    1.184428189982887,
+    1.1124069169789803,
+    1.1181893210117222,
+    1.1042738551018285,
+    1.1043454048012484,
+    1.1286612733238552,
+    1.1188012444171584,
+    1.1209337386597438,
+    1.10587375471238,
+    1.098651764773521,
+    1.1141460502848906,
+    1.0955017307559316,
+    1.098749214277214,
+    1.163088559353051,
+    1.1914743123268212,
+    1.1610766411328517,
+    1.1856480078036045,
+    1.1483205527651543,
+    1.1936752973985272,
+    1.1101547958804112,
+    1.1413028303648578,
+    1.1837641134732912,
+    1.1760416502545195,
+    1.1337165409693866,
+    1.1857522395478577,
+    1.2191999538772915,
+    1.2520315508548618,
+    1.210345393344134,
+    1.2470658045439493,
+    1.2097872178010245,
+    1.2624848476644992,
+    1.3514556236961643,
+    1.33722941210123,
+    1.366594186517037,
+    1.4329219485948734,
+    1.4139539930416423,
+    1.4647649932308358,
+    1.4063804077000177,
+    1.4294493571132505,
+    1.3463562143450023,
+    1.539626619937707,
+    1.4517621866711716,
+    1.5037226854753094,
+    1.4743039648489458,
+    1.4794259386904098,
+    1.5455999228299833,
+    1.695197325502457,
+    1.703609473231126,
+    1.6427013588135315,
+    1.6391914406744372,
+    1.7572424602107841,
+    1.730594052445321,
+    1.7487654256887462,
+    1.8311076000207613,
+    1.9127432956498545,
+    1.8228180259048772,
+    1.8946095726319723,
+    1.842022866761985
   ],
   "auc_roc_macro": [
-    0.5861148917690563,
-    0.6382188769738982,
-    0.6656913099349121,
-    0.6833393187588167,
-    0.691015400952043,
-    0.6896549647699163,
-    0.7048594813164274,
-    0.7115320219326879,
-    0.7227428563181881,
-    0.7173999396129064,
-    0.728869866919284,
-    0.7334977765166267,
-    0.7373187805279444,
-    0.7411044472479785,
-    0.7353552637421784,
-    0.7489385943900191,
-    0.7508086739513764,
-    0.7543862557007509,
-    0.7617968607450436,
-    0.7595833627258213,
-    0.7682776519450237,
-    0.7684956206684662,
-    0.7680464111518411,
-    0.7686820019336239,
-    0.7755240023125906,
-    0.7746318906988712,
-    0.778230769313696,
-    0.7775286541221039,
-    0.7780248975111342,
-    0.7785490184634024,
-    0.7789283074871374,
-    0.7794676814579019,
-    0.7808842283095861,
-    0.7822094763488358,
-    0.7854865314406501,
-    0.7814432103799396,
-    0.7789131089108106,
-    0.783267133597504,
-    0.7850271113827677,
-    0.7887643755934152,
-    0.7877825950780236,
-    0.789577471144223,
-    0.7874829195929393,
-    0.7891315315468567,
-    0.7897503062057237,
-    0.7892659592790113,
-    0.7892256241254351,
-    0.7877708596418972,
-    0.7921449724646248,
-    0.7918404951779995,
-    0.7925665694176522,
-    0.7905315164071969,
-    0.7924461372690477,
-    0.7962187209618695,
-    0.7951120553052311,
-    0.7949196487770661,
-    0.7949877817779832,
-    0.7962038385857333,
-    0.7954472904014142,
-    0.7951010309966685,
-    0.7952121970969696,
-    0.7938179343640532,
-    0.795896438923343,
-    0.7964101759145408,
-    0.7975779480543309,
-    0.794828598633565,
-    0.7924029001696348,
-    0.7954328904682788,
-    0.795619605114598,
-    0.795366641930549,
-    0.7918652655652789,
-    0.7923138588614913,
-    0.795054493956137,
-    0.7964765288681199,
-    0.7947400510084304,
-    0.7946282218425195,
-    0.7947766479794367,
-    0.7952097440377667,
-    0.795785552486931,
-    0.7924229672669126
+    0.5600889783341848,
+    0.6006936068308715,
+    0.6135082712869522,
+    0.619662650828645,
+    0.6503016827609143,
+    0.669356745858073,
+    0.6801516399253448,
+    0.6956458946755157,
+    0.6911901512890865,
+    0.6878028916584179,
+    0.711056476048291,
+    0.7249608742542427,
+    0.7285883097397269,
+    0.7313872465588427,
+    0.7410520255477292,
+    0.7475906933947389,
+    0.7534237658041478,
+    0.7438914668774912,
+    0.7640327407828283,
+    0.7626778933313453,
+    0.7681162343266192,
+    0.7674014927002437,
+    0.769074441384282,
+    0.764215790181779,
+    0.7714702353734016,
+    0.7735906794011821,
+    0.7778793373631595,
+    0.7778735125103365,
+    0.7802085311640891,
+    0.7803372778518832,
+    0.7842598628917763,
+    0.7823228231152382,
+    0.7838151999323013,
+    0.7838244382870435,
+    0.7850657680692592,
+    0.7835648581099827,
+    0.7861069970089803,
+    0.7871647223831422,
+    0.790835876576403,
+    0.7940458179780945,
+    0.7915442563521022,
+    0.7958718966986681,
+    0.7914044080104288,
+    0.7919859158077863,
+    0.7950329793603457,
+    0.7943837575081225,
+    0.7944332949980312,
+    0.7933693057425811,
+    0.7951732943980258,
+    0.7963023289821489,
+    0.7952354460544692,
+    0.7973244129856287,
+    0.7939589758854859,
+    0.7983526417614518,
+    0.7977997949640275,
+    0.7962216526625349,
+    0.799182696741477,
+    0.7958033013422208,
+    0.7996982808695525,
+    0.8008184337727041,
+    0.7993454764932675,
+    0.7987675627632589,
+    0.7970890503264659,
+    0.7970806759058202,
+    0.795822651719548,
+    0.7959650228406401,
+    0.79570970273623,
+    0.7964225370744886,
+    0.7963621380965237,
+    0.7969704582418411,
+    0.7922591015985978,
+    0.7940750137887933,
+    0.7950192853888799,
+    0.7953308259817139,
+    0.7979492069497339
   ],
   "binary_auc_roc": [
-    0.6201408551879701,
-    0.6457985226953593,
-    0.6641528202976529,
-    0.6801798061205759,
-    0.6862877913251162,
-    0.6836190131025841,
-    0.6899832382225456,
-    0.6984366339365421,
-    0.6982638274858186,
-    0.6961336685912958,
-    0.7040745284412298,
-    0.7064610113010039,
-    0.711699865228899,
-    0.7087442249361726,
-    0.7140870204873142,
-    0.72244056500313,
-    0.72134099710891,
-    0.7222470542775873,
-    0.7309260117190117,
-    0.7256573761346448,
-    0.7333856739642091,
-    0.7311119859764539,
-    0.7350918014657842,
-    0.7332327593066136,
-    0.7381255240506908,
-    0.7338860226029045,
-    0.7413918405321454,
-    0.7409066768960914,
-    0.739243022575336,
-    0.7400959602513785,
-    0.7422739434992389,
-    0.741888182795605,
-    0.7449046191321171,
-    0.7448972227470668,
-    0.7477245189492023,
-    0.7396731055560524,
-    0.7400593425420569,
-    0.7439251584451364,
-    0.7453496629825946,
-    0.7447742018123833,
-    0.7479454019027537,
-    0.7459644594733013,
-    0.7472925588556134,
-    0.7485339748464633,
-    0.7496620356330936,
-    0.7503695110698109,
-    0.7504027667859278,
-    0.7492418144991111,
-    0.7497101121359214,
-    0.7487516639064702,
-    0.7518381417680812,
-    0.7512169855068956,
-    0.7519432320723394,
-    0.7521833624369878,
-    0.755443767413724,
-    0.7532853117430612,
-    0.7544367383824644,
-    0.7547453973752694,
-    0.7541489797807487,
-    0.7547803621045989,
-    0.7531380004074736,
-    0.7523417963667612,
-    0.7527101027223404,
-    0.7526647158140765,
-    0.7547582289826675,
-    0.7539878842730217,
-    0.7503244883774774,
-    0.753684548436124,
-    0.7525556751678028,
-    0.7545218528437644,
-    0.7542498395768908,
-    0.753217735679646,
-    0.7525300119530065,
-    0.7549070812318074,
-    0.7539543203741943,
-    0.7530687713640659,
-    0.7530422116177486,
-    0.753277943374621,
-    0.7527464402655615,
-    0.7504382638308479
+    0.5746178702490273,
+    0.6266008971142668,
+    0.6405693344979279,
+    0.6285069511450837,
+    0.6645732655645158,
+    0.6639082072753981,
+    0.6745341790315174,
+    0.683820480545378,
+    0.6845977733743137,
+    0.6847907798008759,
+    0.7005820058503164,
+    0.7125628048347254,
+    0.7135322916085323,
+    0.718656025285775,
+    0.7183330217886297,
+    0.7249310735359807,
+    0.7287012687489955,
+    0.7258562380327049,
+    0.7321114505230029,
+    0.7305286521388216,
+    0.7338493208437527,
+    0.7371632095290542,
+    0.7389203272609013,
+    0.7377104019397357,
+    0.7387708026131203,
+    0.7399987986477615,
+    0.7399050270539593,
+    0.7408331893279577,
+    0.741698118113097,
+    0.7423372050047079,
+    0.7426266165864159,
+    0.7454694620071233,
+    0.7440101048067762,
+    0.7451856817640333,
+    0.7447040482208444,
+    0.7421381469903773,
+    0.7459554101082584,
+    0.747642066065856,
+    0.746851773529862,
+    0.749660242570051,
+    0.7486064818332455,
+    0.7518899444800442,
+    0.7499557337561376,
+    0.7507170571173508,
+    0.7533862555890336,
+    0.7533552692183298,
+    0.7516668762309098,
+    0.7501508974616726,
+    0.7535492282096333,
+    0.754096000371164,
+    0.7529061349427867,
+    0.7552340070543583,
+    0.7526605133225704,
+    0.7547960233896108,
+    0.7533051755195792,
+    0.7546422962503241,
+    0.7534207720526022,
+    0.7545340120525215,
+    0.7559255410399631,
+    0.7548512161113886,
+    0.7570695152610959,
+    0.7554793204918642,
+    0.7527130724830046,
+    0.7546512055323167,
+    0.7543438633201833,
+    0.7555772385439521,
+    0.7555056841219112,
+    0.7562786904095109,
+    0.7549175314273522,
+    0.7555050397398804,
+    0.7539582146829898,
+    0.7558435084057675,
+    0.7558688074046331,
+    0.7562037739942654,
+    0.7562586865499428
   ],
   "binary_f1": [
-    0.29342674634246857,
-    0.3886562618956985,
-    0.5546908164709212,
-    0.5068807339449541,
-    0.4517304189435337,
-    0.4761573268360599,
-    0.5009301538981904,
-    0.5443298969072164,
-    0.5097177623795842,
-    0.5396571062329755,
-    0.5200881505339888,
-    0.5153688524590164,
-    0.535511837279093,
-    0.508598228243877,
-    0.5334668669671173,
-    0.5582309582309583,
-    0.5356721200877045,
-    0.5398274987316083,
-    0.5958029197080292,
-    0.591600730371272,
-    0.5885490196078431,
-    0.5927406951707167,
-    0.592707045735476,
-    0.5736657078938958,
-    0.5940777297964219,
-    0.5775280898876405,
-    0.5820003231539829,
-    0.5931691818903893,
-    0.5983379501385041,
-    0.5910224438902744,
-    0.595234387167108,
-    0.5876913265306123,
-    0.5747502865564107,
-    0.6026377036462374,
-    0.5670701376222849,
-    0.6068195534097767,
-    0.5910513141426783,
-    0.6182739603517663,
-    0.5945006321112516,
-    0.59044908285895,
-    0.5982852689010133,
-    0.6232765011119348,
-    0.5953282828282829,
-    0.629516760992599,
-    0.61021094466524,
-    0.6241364104071733,
-    0.6013071895424836,
-    0.6220180767521114,
-    0.6018779342723005,
-    0.6162422781377128,
-    0.6258443465491924,
-    0.5986159169550173,
-    0.6055900621118012,
-    0.6160808568411525,
-    0.609396805706311,
-    0.6157583118263246,
-    0.6115473441108545,
-    0.6080347448425625,
-    0.619126257318721,
-    0.6191698113207548,
-    0.6112567463377024,
-    0.6089763295419612,
-    0.6293419317015976,
-    0.6164383561643836,
-    0.625821876867902,
-    0.620253164556962,
-    0.6139605462822458,
-    0.6196420514363062,
-    0.6068035550107264,
-    0.6216377764494919,
-    0.6178468971765061,
-    0.6016949152542372,
-    0.6070050918068199,
-    0.6155012892461702,
-    0.6128786725529,
-    0.6181653863533665,
-    0.6215776232055646,
-    0.6236463432725114,
-    0.6278659611992945,
-    0.619210722263589
+    0.0,
+    0.3076923076923077,
+    0.4815471394037067,
+    0.5027052094605039,
+    0.4215361999264976,
+    0.44010743061772606,
+    0.4845203857215361,
+    0.5391762155466089,
+    0.5097206338833524,
+    0.546183500385505,
+    0.5507064364207221,
+    0.5256604408547871,
+    0.5531776913099871,
+    0.5504140282513394,
+    0.588181955568989,
+    0.6085943190094684,
+    0.5395756146850792,
+    0.5147649834003145,
+    0.6209262435677531,
+    0.5954036891442395,
+    0.5705930138099107,
+    0.5855181023720349,
+    0.5922572588198564,
+    0.5806146572104018,
+    0.5733525733525734,
+    0.6051437216338881,
+    0.6163150907213454,
+    0.5778063272844066,
+    0.6007314843035659,
+    0.595968992248062,
+    0.5999690258634041,
+    0.6026936026936027,
+    0.5788623141564319,
+    0.6115702479338843,
+    0.6026469682979378,
+    0.6046583371895727,
+    0.6219459303166113,
+    0.6203911031497238,
+    0.619626445300919,
+    0.6015343666823235,
+    0.599250936329588,
+    0.6206586826347306,
+    0.6065015479876161,
+    0.6278079577154603,
+    0.6099643023436288,
+    0.6168337891218475,
+    0.6127114769090077,
+    0.6244976931090936,
+    0.6102374344742523,
+    0.6280250971018823,
+    0.629918938835667,
+    0.616816725164047,
+    0.6197183098591549,
+    0.6256611757594076,
+    0.617359413202934,
+    0.6237533998186764,
+    0.6274450797472164,
+    0.6245656443571537,
+    0.6238088035093027,
+    0.6288444311734847,
+    0.6219124109713593,
+    0.6217297630040012,
+    0.6473523703909065,
+    0.6262778111846061,
+    0.6288288288288288,
+    0.6338600451467269,
+    0.6291460303166742,
+    0.6284259397933204,
+    0.6229258639062262,
+    0.6284251731406203,
+    0.6075869667873446,
+    0.6178736517719569,
+    0.626969696969697,
+    0.6349442379182156,
+    0.6363636363636364
   ],
   "learning_rate": [
     0.0009997532801828658,
@@ -484,11 +459,6 @@
     0.0001812880051256552,
     0.00016934406733817422,
     0.0001577264470356557,
-    0.00014644660940672634,
-    0.0001355156862892944,
-    0.00012494446518477025,
-    0.00011474337861210548,
-    0.00010492249381215483,
-    9.549150281252637e-05
+    0.00014644660940672634
   ]
 }

README.md

@@ -1,105 +1,157 @@
----
-license: mit
-language:
-- en
-library_name: pytorch
-pipeline_tag: image-classification
-tags:
-- chexvision
-- medical-imaging
-- chest-xray
-- radiology
-- pytorch
-- multi-label-classification
-datasets:
-- HlexNC/chest-xray-14
----
-
-# CheXVision-ResNet
-
-> **CheXVision** — Deep Learning & Big Data university project.
-> 14-class chest X-ray pathology detection + binary normal/abnormal classification
-> on the NIH Chest X-ray14 dataset (112,120 images).
-
-## Architecture
-
-<p align="center">
-  <img src="https://huggingface.co/HlexNC/chexvision-scratch/resolve/main/arch_scratch.png" width="88%" alt="SE-ResNet Architecture"/>
-</p>
-
-## Training Pipeline
-
-<p align="center">
-  <img src="https://huggingface.co/HlexNC/chexvision-scratch/resolve/main/pipeline_training.png" width="42%" alt="Training Pipeline"/>
-</p>
-
-## Training Metrics
-
-- Best validation macro AUC-ROC: `0.7976`
-- Best validation binary AUC-ROC: `0.7554`
-- Best validation binary F1: `0.6295`
-- Best checkpoint epoch: `65`
-
-
-## Per-Class AUC-ROC at Best Epoch
-
-| Pathology            | AUC-ROC  | Visual        |
-|----------------------|----------|---------------|
-| Atelectasis          | `0.7842` | `████████░░` |
-| Cardiomegaly         | `0.8929` | `█████████░` |
-| Effusion             | `0.8602` | `█████████░` |
-| Infiltration         | `0.6903` | `███████░░░` |
-| Mass                 | `0.8168` | `████████░░` |
-| Nodule               | `0.6634` | `███████░░░` |
-| Pneumonia            | `0.6807` | `███████░░░` |
-| Pneumothorax         | `0.8208` | `████████░░` |
-| Consolidation        | `0.8184` | `████████░░` |
-| Edema                | `0.9103` | `█████████░` |
-| Emphysema            | `0.8355` | `████████░░` |
-| Fibrosis             | `0.7604` | `████████░░` |
-| Pleural_Thickening   | `0.7702` | `████████░░` |
-| Hernia               | `0.8619` | `█████████░` |
-
-## Training Configuration
-
-- Repository: `HlexNC/chexvision-scratch`
-- Dataset: [HlexNC/chest-xray-14](https://huggingface.co/datasets/HlexNC/chest-xray-14) · revision `c4e9a86b`
-- Architecture: Custom residual CNN with Squeeze-Excitation channel attention (depth [3, 4, 6, 3]) trained from scratch with shared features and dual classification heads.
-- Platform: Kaggle GPU kernel (NVIDIA T4 / P100)
-- Batch size: `32` × grad_accum `4` = **effective batch `128`**
-- AMP (fp16): `enabled`
-- Optimizer: AdamW  ·  Scheduler: CosineAnnealingLR
-- Epochs configured: `100`  ·  Early stop patience: `15`
-
-## Intended Use
-
-This model is intended for research and educational work on automated chest X-ray pathology detection.
-It outputs two predictions per image:
-1. **Multi-label scores** — independent sigmoid probability for each of 14 NIH pathologies
-2. **Binary score** — sigmoid probability of any abnormality (Normal vs. Abnormal)
-
-## Limitations
-
-- Not validated for clinical use. Predictions must not substitute professional medical judgment.
-- Trained on NIH Chest X-ray14, which contains noisy radiologist annotations (patient-level labels, not lesion-level).
-- Performance degrades on images from equipment, patient populations, or preprocessing pipelines
-  that differ from the NIH training distribution.
-- Reported AUC metrics are on the validation split, not the held-out test set.
-
-## CheXNet Benchmark Context
-
-CheXNet (Rajpurkar et al., 2017) — the seminal paper establishing DenseNet-121 for chest X-ray
-classification — reported **0.841 macro AUC-ROC** on a comparable split of this dataset.
-CheXVision-DenseNet matches this benchmark. See the
-[CheXVision demo](https://huggingface.co/spaces/HlexNC/chexvision-demo) for live inference.
-
-## Citation
-
-```bibtex
-@misc{chexvision2026,
-  title={CheXVision: Dual-Task Chest X-ray Classification with Custom CNN and DenseNet-121},
-  author={BIG D(ATA) Team},
-  year={2026},
-  howpublished={\url{https://huggingface.co/HlexNC/chexvision-scratch}}
-}
-```
+---
+license: mit
+language:
+- en
+library_name: pytorch
+pipeline_tag: image-classification
+tags:
+- chexvision
+- medical-imaging
+- chest-xray
+- radiology
+- pytorch
+- multi-label-classification
+datasets:
+- HlexNC/chest-xray-14-320
+---
+
+# CheXVision-ResNet
+
+> **CheXVision** — Deep Learning & Big Data university project.
+> 14-class chest X-ray pathology detection + binary normal/abnormal classification
+> on the NIH Chest X-ray14 dataset (112,120 images).
+
+## Architecture
+
+```mermaid
+graph LR
+    IN["Input
+    3 × 224 × 224"] --> STEM["Stem
+    7×7 Conv · BN · ReLU
+    3→64ch · MaxPool ÷2"]
+    STEM --> S1["Stage 1
+    3× SE-ResBlock
+    64ch"]
+    S1 --> S2["Stage 2 ↓½
+    4× SE-ResBlock
+    128ch"]
+    S2 --> S3["Stage 3 ↓½
+    6× SE-ResBlock
+    256ch"]
+    S3 --> S4["Stage 4 ↓½
+    3× SE-ResBlock
+    512ch"]
+    S4 --> GAP["Global Avg Pool
+    Dropout(0.5)
+    512-dim"]
+    GAP --> MLH["Multilabel Head
+    Linear 512→14
+    sigmoid · 14 pathologies"]
+    GAP --> BH["Binary Head
+    Linear 512→1
+    sigmoid · Normal/Abnormal"]
+    style MLH fill:#2e7d32,color:#fff
+    style BH fill:#1565c0,color:#fff
+    style IN fill:#37474f,color:#fff
+```
+
+## Training Pipeline
+
+```mermaid
+flowchart TD
+    DS[("🗄️ HlexNC/chest-xray-14
+112,120 images · 36 shards · ~4.7 GB")]
+    DS -->|snapshot_download| PREP["📂 data/images · data/labels.csv
+train 78,468 · val 11,210 · test 22,442"]
+    PREP --> AUG["Augmentation Pipeline
+HFlip · Rotate±15° · RandomAffine
+ColorJitter · GaussianBlur · RandomErasing"]
+    AUG --> FWD["⚡ Model Forward Pass
+torch.cuda.amp.autocast · fp16"]
+    FWD --> ML["multilabel_logits B×14
+WeightedBCE + pos_weight · 14 classes"]
+    FWD --> BIN["binary_logits B×1
+BCE · Normal vs. Abnormal"]
+    ML --> LOSS["Combined Loss
+1.0 × multilabel + 0.5 × binary"]
+    BIN --> LOSS
+    LOSS --> BACK["Backward · Grad Clip 1.0
+Gradient Accumulation ×4 · eff. batch 128"]
+    BACK --> OPT["AdamW · CosineAnnealingLR
+early stop patience = 15"]
+    OPT -->|"↑ best val AUC-ROC"| BEST["💾 Best Checkpoint
+model_state · best_val_metrics · config"]
+    BEST -->|upload_model_artifacts| HUB["🤗 HF Hub
+checkpoint · history.json · model card"]
+```
+
+## Training Metrics
+
+- Best validation macro AUC-ROC: `0.8008`
+- Best validation binary AUC-ROC: `0.7571`
+- Best validation binary F1: `0.6474`
+- Best checkpoint epoch: `60`
+
+
+## Per-Class AUC-ROC at Best Epoch
+
+| Pathology            | AUC-ROC  | Visual        |
+|----------------------|----------|---------------|
+| Atelectasis          | `0.7841` | `████████░░` |
+| Cardiomegaly         | `0.8985` | `█████████░` |
+| Effusion             | `0.8623` | `█████████░` |
+| Infiltration         | `0.6872` | `███████░░░` |
+| Mass                 | `0.8322` | `████████░░` |
+| Nodule               | `0.6899` | `███████░░░` |
+| Pneumonia            | `0.6741` | `███████░░░` |
+| Pneumothorax         | `0.8183` | `████████░░` |
+| Consolidation        | `0.8110` | `████████░░` |
+| Edema                | `0.9151` | `█████████░` |
+| Emphysema            | `0.8267` | `████████░░` |
+| Fibrosis             | `0.7579` | `████████░░` |
+| Pleural_Thickening   | `0.7752` | `████████░░` |
+| Hernia               | `0.8791` | `█████████░` |
+
+## Training Configuration
+
+- Repository: `HlexNC/chexvision-scratch`
+- Dataset: [HlexNC/chest-xray-14-320](https://huggingface.co/datasets/HlexNC/chest-xray-14-320) · revision `44443e6ee968b3c6094b63f14a27698c40b50680`
+- Architecture: Custom residual CNN with Squeeze-Excitation channel attention (depth [3, 4, 6, 3]) trained from scratch with shared features and dual classification heads.
+- Platform: Kaggle GPU kernel (NVIDIA T4 / P100)
+- Batch size: `24` × grad_accum `4` = **effective batch `96`**
+- AMP (fp16): `enabled`
+- Optimizer: AdamW  ·  Scheduler: CosineAnnealingLR
+- Epochs configured: `100`  ·  Early stop patience: `15`
+
+## Intended Use
+
+This model is intended for research and educational work on automated chest X-ray pathology detection.
+It outputs two predictions per image:
+1. **Multi-label scores** — independent sigmoid probability for each of 14 NIH pathologies
+2. **Binary score** — sigmoid probability of any abnormality (Normal vs. Abnormal)
+
+## Limitations
+
+- Not validated for clinical use. Predictions must not substitute professional medical judgment.
+- Trained on NIH Chest X-ray14, which contains noisy radiologist annotations (patient-level labels, not lesion-level).
+- Performance degrades on images from equipment, patient populations, or preprocessing pipelines
+  that differ from the NIH training distribution.
+- Reported AUC metrics are on the validation split, not the held-out test set.
+
+## CheXNet Benchmark Context
+
+CheXNet (Rajpurkar et al., 2017) — the seminal paper establishing DenseNet-121 for chest X-ray
+classification — reported **0.841 macro AUC-ROC** on a comparable split of this dataset.
+CheXVision-DenseNet matches this benchmark. See the
+[CheXVision demo](https://huggingface.co/spaces/HlexNC/chexvision-demo) for live inference.
+
+## Citation
+
+```bibtex
+@misc{chexvision2026,
+  title={CheXVision: Dual-Task Chest X-ray Classification with Custom CNN and DenseNet-121},
+  author={BIG D(ATA) Team},
+  year={2026},
+  howpublished={\url{https://huggingface.co/HlexNC/chexvision-scratch}}
+}
+```

training_config.json

@@ -1,8 +1,8 @@
 {
   "seed": 42,
   "data": {
-    "dataset_name": "HlexNC/chest-xray-14",
-    "image_size": 224,
+    "dataset_name": "HlexNC/chest-xray-14-320",
+    "image_size": 320,
     "num_workers": 4,
     "pin_memory": true,
     "train_split": 0.7,
@@ -38,12 +38,12 @@
       "Hernia"
     ],
     "data_dir": "/kaggle/working/data",
-    "hf_dataset_repo": "HlexNC/chest-xray-14",
-    "hf_dataset_revision": "c4e9a86b38de3b1604afa6e9f514d156eb9d20bf"
+    "hf_dataset_repo": "HlexNC/chest-xray-14-320",
+    "hf_dataset_revision": "44443e6ee968b3c6094b63f14a27698c40b50680"
   },
   "training": {
     "epochs": 100,
-    "batch_size": 32,
+    "batch_size": 24,
     "learning_rate": 0.001,
     "weight_decay": 0.0001,
     "optimizer": "adamw",
@@ -67,7 +67,7 @@
     "log_every_n_steps": 100
   },
   "huggingface": {
-    "dataset_repo": "HlexNC/chest-xray-14",
+    "dataset_repo": "HlexNC/chest-xray-14-320",
     "scratch_model_repo": "HlexNC/chexvision-scratch",
     "transfer_model_repo": "HlexNC/chexvision-densenet",
     "space_repo": "HlexNC/chexvision-demo"